1. 程式人生 > >舞蹈鏈——精確覆蓋 模板

舞蹈鏈——精確覆蓋 模板

這篇部落格很精細:點我點我!!

D a n c i n g L

i n k s
Dancing Links(精確覆蓋模板):

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

#define N 500010
#define M 1010
struct DancingLink{
    int n,s,ansd;//列數 節點總數 
    int S[M],A[M],H[M];//S[]該列節點總數  A[]答案  H[]行首指標 
    int L[N],R[N],U[N],D[N]; //L[],R[],U[],D[] 上下左右 
    int X[N],C[N];//X[] C[] 行列編號 
    void init(int n){//初始化 
        this->n=n;
        for(int i=0;i<=n;i++)
            U[i]=i,D[i]=i,L[i]=i-1,R[i]=i+1;
        R[n]=0,L[0]=n;s=n+1;
		memset(S,0,sizeof(S));
		memset(H,-1,sizeof(H));
    }
    void DelCol(int c){//刪除列 
        L[R[c]]=L[c];R[L[c]]=R[c];
        for(int i=D[c];i!=c;i=D[i])
            for(int j=R[i];j!=i;j=R[j])
                U[D[j]]=U[j],D[U[j]]=D[j],--S[C[j]];
    }
    void ResCol(int c){//恢復列 
        for(int i=U[c];i!=c;i=U[i])
            for(int j=L[i];j!=i;j=L[j])
                ++S[C[j]],U[D[j]]=j,D[U[j]]=j;
        L[R[c]]=c,R[L[c]]=c;
    }
    void AddNode(int r,int c){//新增節點 
        ++S[c],C[++s]=c,X[s]=r;
        D[s]=D[c],U[D[c]]=s,U[s]=c,D[c]=s;
        if(H[r]<0) H[r]=L[s]=R[s]=s;//行首節點
        else  R[s]=R[H[r]],L[R[H[r]]]=s,L[s]=H[r],R[H[r]]=s;
    }
    bool dfs(int d){//深度,深搜遍歷 
        if(!R[0]){
            ansd=d;return true;
        }
        int c=R[0];
        for(int i=R[0];i;i=R[i]) if(S[i]<S[c]) c=i;
        DelCol(c);
        for(int i=D[c];i!=c;i=D[i]){
            A[d]=X[i];
            for(int j=R[i];j!=i;j=R[j]) DelCol(C[j]);
            if(dfs(d+1)) return true;
            for(int j=L[i];j!=i;j=L[j]) ResCol(C[j]);
        }
        ResCol(c);return false;
    }

} dlx; 

int main(){
	int n,m;
	while(scanf("%d%d",&n,&m)!=EOF) {
		dlx.init(m);
		for(int i=1; i<=n; i++) {
			int num,j;
			scanf("%d",&num);	//第 i行有 num個 1
			while(num--) {
				scanf("%d",&j);
				dlx.AddNode(i,j);
			}
		}
		if(!dlx.dfs(0)) printf("NO\n");
		else {
			printf("%d",dlx.ansd);
			sort(dlx.A,dlx.A+dlx.ansd);
			for(int i=0; i<dlx.ansd; i++) 
				printf(" %d",dlx.A[i]);
			printf("\n");
		}
	}
}