HDU 5113 DFS + 剪枝

HDU 5113
题目链接：
http://acm.hdu.edu.cn/showproblem.php?pid=5113
题意：
给一个矩阵，一些颜色和每种颜色的使用次数。问一种合法的方案使得填色后相邻格子颜色不同。
没有合法方案输出impossible。
思路：
卡常数，用结构体会T。
剪枝，如果一种颜色的个数大于剩余格子个数的一半就减掉。
自己写的时候每次都对颜色按剩余数排序……其实直接从前往后遍历找就好啊，排序会T啊……
源码：

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <iostream>
#include <string>
using namespace std;
const int MAXN = 5 + 1;
int gra[MAXN][MAXN];
int cnt[MAXN*MAXN];
int n, m, l;
bool valid(int x, int y)
{
    if(x > 1 && gra[x][y] == gra[x - 1][y]) return false;
    if(y > 1 && gra[x][y] == gra[x][y - 1]) return false;
    return true;
}
bool dfs(int x, int y, int mark, int rest)
{
    gra[x][y] = mark;
    if(!valid(x, y))    return false;
    for(int i = 1 ; i <= l ; i++)
        if(cnt[i] > (rest + 1) / 2) return false;
    cnt[mark]--;
    if(x == n && y == m)    return true;
    int ty = y == m ? 1 : y + 1;
    int tx = y == m ? x + 1 : x;
    for(int i = 1 ; i <= l ; i++){
        if(cnt[i] != 0 && dfs(tx, ty, i, rest - 1)){
            return true;
        }
    }
    cnt[mark]++;
    return false;
}
int main()
{
    int t;
    scanf("%d", &t);
    for(int cas = 1 ; cas <= t ; cas++){
        scanf("%d%d%d", &n, &m, &l);
        for(int i = 1 ; i <= l ; i++)
            scanf("%d", &cnt[i]);
        int ok = 0;
        for(int i = 1 ; i <= l ; i++){
            if(dfs(1, 1, i, n * m)){
                ok = 1;
                break;
            }
        }
        printf("Case #%d:\n", cas);
        if(ok){
            printf("YES\n");
            for(int i = 1 ; i <= n ; i++){
                int f = 1;
                for(int j = 1 ; j <= m ; j++){
                    if(f)   f = 0;
                    else    printf(" ");
                    printf("%d", gra[i][j]);
                }
                printf("\n");
            }
        }
        else    printf("NO\n");
    }
    return 0;
}