USACO 6.4.1 Wisconsin Squares 暴力

http://train.usaco.org/usacoprob2?a=VNPgIbuqM60&S=wissqu

题目大意：一个4*4的牧场有ABCDE五种奶牛，其中C4个别的各3个，现在要用另一批奶牛替换这一批，其中D4个别的各3个，要求用来替换的奶牛种类和当前种类不同并且和周围八个格子的种类也不同（原本的奶牛分布同样满足这个条件），并且第一个用于替换的是D种类，要求输出字典序最小的替换顺序以及能够和这个顺序的替换等效的所有可能情况总数

时限：5秒

时限很宽松，暴力枚举就行。如果已经超过字典序且和已知答案不匹配了就剪枝

/*
ID: frontie1
TASK: wissqu
LANG: C++
*/
#include <iostream>
#include <cstdio>
#include <cstring>

using namespace std;

char ori[6][6], arr[6][6], ans[6][6];
int remain[5] = {3, 3, 3, 4, 3};
bool yearling[6][6];

int go_r[8] = {0, -1, -1, -1, 0, 1, 1, 1};
int go_c[8] = {1, 1, 0, -1, -1, -1, 0, 1};

string output_str = "ZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZ";
int cnt = 0;

bool valid(int r, int c, char chr)
{
    if(arr[r][c] == chr) return false;
    int nxt_r, nxt_c;
    for(int i = 0; i < 8; ++i){
        nxt_r = r + go_r[i];
        nxt_c = c + go_c[i];
        if(nxt_r > 4 || nxt_r <= 0 || nxt_c > 4 || nxt_c <= 0) continue;
        else if(arr[nxt_r][nxt_c] == chr) return false;
    }
    return true;
}

bool same(char arr[][6], char ans[][6])
{
    for(int i = 1; i < 5; ++i){
        for(int k = 1; k < 5; ++k){
            if(arr[i][k] != ans[i][k]) return false;
        }
    }
    return true;
}

void judge(string log)
{
    if(log < output_str){
        output_str = log;
        cnt = 1;
        for(int i = 1; i <= 4; ++i){
            for(int k = 1; k <= 4; ++k){
                ans[i][k] = arr[i][k];
            }
        }
    }
    else if(same(arr, ans)) ++cnt;
}

bool match(string str)
{
    int len = str.length();
    if(len < 3) return true;
    if(ans[str[len-2]-'0'][str[len-1]-'0'] == str[len-3]) return true;
    else return false;
}

void DFS(int dep, char chr, string log)
{
    if(log > output_str && !match(log)) return;

    --remain[chr-'A'];
    for(int i = 1; i <= 4; ++i){
        for(int k = 1; k <= 4; ++k){
            if(yearling[i][k]) continue;
            if(valid(i, k, chr)){
                yearling[i][k] = true;
                arr[i][k] = chr;
                string nxt_log = (((log+chr)+(char)('0'+i))+(char)('0'+k));
                if(dep == 15) judge(nxt_log);
                else{
                    for(int i = 0; i < 5; ++i){
                        if(remain[i] > 0) DFS(dep+1, 'A'+i, nxt_log);
                    }
                }
                arr[i][k] = ori[i][k];
                yearling[i][k] = false;
            }
        }
    }
    ++remain[chr-'A'];
}

int main()
{
    freopen("wissqu.in", "r", stdin);
    freopen("wissqu.out", "w", stdout);

    for(int i = 1; i <= 4; ++i){
        cin >> ori[i]+1;
        for(int k = 1; k <= 4 ; ++k){
            arr[i][k] = ori[i][k];
        }
    }

    DFS(0, 'D', "");

    for(int i = 0; i < 16; ++i){
        cout << output_str[i*3] << ' ' << output_str[i*3+1] << ' ' << output_str[i*3+2] << endl;
    }
    cout << cnt << endl;

    return 0;
}