hdu5113 Black And White DFS+剪枝

点击打开链接

题意：

给你n*m的方格，你要用k种颜色去填，每种颜色要填ci次，并且要保证没有任意两个相邻的格子有相同的颜色，构造一种可行方法。

思路：

直接暴力回溯+剪枝，当没填的颜色中同种颜色数量超过(剩余格子+1)/2的时候，不可能合法。

代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define mem(a) memset(a,0,sizeof(a))
#define mp(x,y) make_pair(x,y)
const int INF = 0x3f3f3f3f;
const ll INFLL = 0x3f3f3f3f3f3f3f3fLL;
inline ll read(){
    ll x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
//////////////////////////////////////////////////////////////////////////
const int maxn = 1e5+10;

int n,m,k,c[30];
int mp[10][10],flag;
int dx[4] = {0,0,1,-1};
int dy[4] = {1,-1,0,0};

bool ok(int x,int y,int col){
    for(int i=0; i<4; i++){
        int tx=x+dx[i],ty=y+dy[i];
        if(tx<0 || tx>=n || ty<0 || ty>=m) continue;
        if(mp[tx][ty] == col) return false;
    }
    return true;
}

void dfs(int sta){
    if(flag) return ;
    if(sta == n*m){
        flag = true;
        puts("YES");
        for(int i=0; i<n; i++){
            for(int j=0; j<m; j++){
                cout << mp[i][j];
                if(j==m-1) cout << endl;
                else cout << " ";
            }
        }
        return ;
    }

    int ma=-1;
    for(int i=1; i<=k; i++) ma = max(ma,c[i]);
    if(ma>(n*m-sta+1)/2) return ;

    int x=sta/m,y=sta%m;
    for(int i=1; i<=k; i++){
        if(c[i]==0) continue;
        if(ok(x,y,i)){
            mp[x][y] = i;
            c[i]--;
            dfs(sta+1);
            mp[x][y] = -1;
            c[i]++;
        }
    }
}

int main(){
    int T=read();
    for(int cas=1; cas<=T; cas++){
        memset(mp,-1,sizeof(mp));
        flag = 0;
        n=read(),m=read(),k=read();
        for(int i=1; i<=k; i++)
            c[i] = read();
        cout << "Case #" << cas << ":
";
        dfs(0);
        if(!flag) puts("NO");
    }

    return 0;
}