UVa 1603 破坏正方形

https://vjudge.net/problem/UVA-1603

题意：有一个火柴棍组成的正方形网格，计算至少要拿走多少根火柴才能破坏所有正方形。

思路：从边长为1的正方形开始遍历，将正方形的边长和它的实际火柴数保存起来。之后dfs搜索。

#include<iostream>
#include<cstring>
using namespace std;

const int maxn = ;

int n, m,best,s;
int map[maxn],size[maxn],fullsize[maxn];
int contins[maxn][maxn];

int row_match(int x, int y)   //行
{
    return ( * n + )*x + y;
}

int col_match(int x, int y)  //列
{
    return ( * n + )*x + n + y;
}

int edges()  //计算火柴根数
{
    return *n*(n + );
}

void init()
{
    int ans;
    cin >> n >> m;
    //初始化网格
    for (int i = ; i < edges(); i++)
        map[i] = ;
    while (m--)
    {
        cin >> ans;
        map[ans - ] = ;
    }
    s = ;  //s用来为正方形编号
    memset(contins, , sizeof(contins));
    for (int i = ; i <= n;i++)  //正方形的边长，从1开始到n
    for (int x = ; x <= n - i;x++) //正方形左上角的x坐标
    for (int y = ; y <= n - i; y++) //正方形左上角的y坐标
    {
        size[s] = ;
        fullsize[s] =  * i;  //第s个正方形的边长
        //cout << fullsize[s] << "*" << endl;
        for (int j = ; j < i; j++)
        {
            int a = row_match(x, y + j);
            int b = row_match(x + i, y + j);
            int c = col_match(x + j, y);
            int d = col_match(x + j, y + i);
            contins[s][a] = ;  //储存正方形的边
            contins[s][b] = ;
            contins[s][c] = ;
            contins[s][d] = ;
            size[s] += map[a] + map[b] + map[c] + map[d];
        }
        s++;
    }
}

int find_sqware()
{
    for (int i = ; i < s;i++)
    if (fullsize[i] == size[i])     return i; //找到未被破坏的正方形
    return -;
}

void dfs(int dep)
{
    if (dep >= best) return;
    int k = find_sqware();
    if (k == -) //每个正方形都已经被破坏
    {
        best = dep;
        return;
    }
    for (int i = ; i < edges(); i++)
    {
        if (contins[k][i])
        {
            for (int j = ; j < s;j++)
            if (contins[j][i])    size[j]--;  //破坏，边的数量减少
            dfs(dep + );
            for (int j = ; j < s;j++)  //回溯
            if (contins[j][i])  size[j]++;
        }
    }
}
int main()
{
    //freopen("D:\\txt.txt", "r", stdin);
    int t;
    cin >> t;
    while (t--)
    {
        init();
        best = n*n;
        dfs();
        cout << best << endl;
    }
    return ;
}