hdu 4655 Unshuffle 搜索

http://acm.hdu.edu.cn/showproblem.php?pid=4665

搜索;

搜索前的剪枝：对于某个数num,第一次出现的num标记为0，最后一个出现的num标记为1；

搜索中的剪枝：如果不能匹配，一定是错的

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 2002;
int a[maxn] , t[maxn] , id[maxn] , n , T , pos[maxn][5];int idd[maxn];
int f[maxn] , g[maxn] , cf , cg;
void init() {
    memset(t,0,sizeof(int)*(n+1));
    memset(id , -1 ,sizeof(int)*(n+1));
    memset(idd , -1 ,sizeof(int)*(n+1));
    cf = cg = 0;
}
bool dfs(int p) {
    if(p > n) return true;
    if(id[p] == 0) {
        if(cf < cg && a[p] == g[cf+1] || cf >= cg && cf <= n/2) {
            cf ++;
            f[cf] = a[p];
            idd[p] = 0;
            if(dfs(p+1)) return true;
            cf --;
        }
        return false;
    }
    if(id[p] == 1) {
        if(cg < cf && a[p] == f[cg+1] || cg >= cf && cg <= n/2) {
            cg ++;
            g[cg] = a[p];
            idd[p] = 1;
            if(dfs(p+1)) return true;
            cg --;
        }
        return false;
    }
    if(cf < cg && a[p] == g[cf+1] || cf >= cg && cf <= n/2) {
        cf ++;
        idd[p] = 0;
        f[cf] = a[p];
        if(dfs(p+1)) return true;
        cf --;
    }
    if(cg < cf && a[p] == f[cg+1] || cg >= cf && cg <= n/2) {
        cg ++;
        idd[p] = 1;
        g[cg] = a[p];
        if(dfs(p+1)) return true;
        cg --;
    }
    return false;
}
int main() {
    scanf("%d" , &T);
    while(T--) {
        scanf("%d" , &n);
        for(int i=1;i<=n;i++) scanf("%d",&a[i]);
        init();
        for(int i=1;i<=n;i++) {
            t[a[i]] ++;
            pos[a[i]][t[a[i]]] = i;
        }
        for(int i=1;i<=n;i++) {
            if(t[ a[i] ] == 4) {
                id[ pos[a[i]] [1] ] = 0;
                id[ pos[a[i]] [4] ] = 1;
            }
            if(t[ a[i] ] == 2) {
                id[ pos[a[i]] [1] ] = 0;
                id[ pos[a[i]] [2] ] = 1;
            }
        }
        if(dfs(1)) ;
        for(int i=1;i<=n;i++) printf("%d", idd[i]); puts("");
    }
    return 0;
}