以前的代码太挫了,重新整理dlx,学习HH把精确覆盖,重复覆盖整合在一起。
代码:
struct DLX{ const static int maxn=20010; #define FF(i,A,s) for(int i = A[s];i != s;i = A[i]) int L[maxn],R[maxn],U[maxn],D[maxn]; int size,col[maxn],row[maxn],s[maxn],H[maxn]; bool vis[70]; int ans[maxn],cnt; void init(int m){ for(int i=0;i<=m;i++){ L[i]=i-1;R[i]=i+1;U[i]=D[i]=i;s[i]=0; } memset(H,-1,sizeof(H)); L[0]=m;R[m]=0;size=m+1; } void link(int r,int c){ U[size]=c;D[size]=D[c];U[D[c]]=size;D[c]=size; if(H[r]<0)H[r]=L[size]=R[size]=size; else { L[size]=H[r];R[size]=R[H[r]]; L[R[H[r]]]=size;R[H[r]]=size; } s[c]++;col[size]=c;row[size]=r;size++; } void del(int c){//精确覆盖 L[R[c]]=L[c];R[L[c]]=R[c]; FF(i,D,c)FF(j,R,i)U[D[j]]=U[j],D[U[j]]=D[j],--s[col[j]]; } void add(int c){ //精确覆盖 R[L[c]]=L[R[c]]=c; FF(i,U,c)FF(j,L,i)++s[col[U[D[j]]=D[U[j]]=j]]; } bool dfs(int k){//精确覆盖 if(!R[0]){ cnt=k;return 1; } int c=R[0];FF(i,R,0)if(s[c]>s[i])c=i; del(c); FF(i,D,c){ FF(j,R,i)del(col[j]); ans[k]=row[i];if(dfs(k+1))return true; FF(j,L,i)add(col[j]); } add(c); return 0; } void remove(int c){//重复覆盖 FF(i,D,c)L[R[i]]=L[i],R[L[i]]=R[i]; } void resume(int c){//重复覆盖 FF(i,U,c)L[R[i]]=R[L[i]]=i; } int A(){//估价函数 int res=0; memset(vis,0,sizeof(vis)); FF(i,R,0)if(!vis[i]){ res++;vis[i]=1; FF(j,D,i)FF(k,R,j)vis[col[k]]=1; } return res; } void dfs(int now,int &ans){//重复覆盖 if(R[0]==0)ans=min(ans,now); else if(now+A()<ans){ int temp=INF,c; FF(i,R,0)if(temp>s[i])temp=s[i],c=i; FF(i,D,c){ remove(i);FF(j,R,i)remove(j); dfs(now+1,ans); FF(j,L,i)resume(j);resume(i); } } } }dlx;
最新版dlx模板(精确覆盖+重复覆盖),码迷,mamicode.com
原文地址:http://blog.csdn.net/xianxingwuguan1/article/details/24671115