[DLX] hust 1017 Exact cover

题意：

给你N个包，要拿到M个东西（编号1~M每个只能有一个）

然后每个包里有k个东西，每个东西都有编号。

思路：

舞蹈连模板题

代码：

#include"stdio.h"#include"algorithm"#include"string.h"#include"iostream"#include"queue"#include"map"#include"vector"#include"string"using namespace std;#define N 1010*100 //总共元素个数#define RN 1010//总共行数#define CN 1010//总共列数struct DLX{int n,m,C;//n,m代表范围行列数，C代表元素个数int U[N],D[N],L[N],R[N],Row[N],Col[N]; //U、D、L、R代表第C个的上下左右分别是哪个 Row和Col代表第C的行号和列号int H[RN],S[CN],cnt,ans[RN];//H为每行开头是哪个，H[i]==-1代表本行没有元素 S代表每列有多少个元素 cnt为结果是选了几行 ans存结果是哪几行void init(int _n,int _m)//初始化 创建0~m个原始元素（第0行）{n=_n;m=_m;for(int i=0; i<=m; i++){U[i]=D[i]=i;L[i]=(i==0?m:i-1);R[i]=(i==m?0:i+1);S[i]=0;}C=m;for(int i=1; i<=n; i++) H[i]=-1;}void link(int x,int y)//连接行数，就是把元素插入{C++;Row[C]=x;Col[C]=y;S[y]++;U[C]=U[y];D[C]=y;D[U[y]]=C;U[y]=C;if(H[x]==-1) H[x]=L[C]=R[C]=C;else{L[C]=L[H[x]];R[C]=H[x];R[L[H[x]]]=C;L[H[x]]=C;}}void del(int x)//删除函数，删除对应列{R[L[x]]=R[x];L[R[x]]=L[x];for(int i=D[x]; i!=x; i=D[i]){for(int j=R[i]; j!=i; j=R[j]){U[D[j]]=U[j];D[U[j]]=D[j];S[Col[j]]–;}}}void rec(int x)//恢复函数，恢复对应列{for(int i=U[x]; i!=x; i=U[i]){for(int j=L[i]; j!=i; j=L[j]){U[D[j]]=j;D[U[j]]=j;S[Col[j]]++;}}R[L[x]]=x;L[R[x]]=x;}int dance(int x)//舞蹈连搜索函数{if(R[0]==0)//满足条件，，默认是R[0]=0也就是第0行没有元素了，但是因题目而定{cnt=x;return 1;}int now=R[0];for(int i=R[0]; i!=0; i=R[i])//选择含有元素最多的列，可加速{if(S[i]<S[now]) now=i;}del(now);for(int i=D[now]; i!=now; i=D[i]){ans[x]=Row[i];for(int j=R[i]; j!=i; j=R[j]) del(Col[j]);if(dance(x+1)) return 1;for(int j=L[i]; j!=i; j=L[j]) rec(Col[j]);}rec(now);return 0;}} dlx;int main(){int n,m;while(scanf("%d%d",&n,&m)!=-1){dlx.init(n,m);for(int i=1;i<=n;i++){int k;scanf("%d",&k);while(k–){int x;scanf("%d",&x);dlx.link(i,x);}}int f=dlx.dance(0);if(f==0) puts("NO");else{printf("%d",dlx.cnt);for(int i=0; i<dlx.cnt; i++) printf(" %d",dlx.ans[i]);puts("");}}return 0;}

版权声明：本文为博主原创文章，未经博主允许不得转载。

停止每日在车水马龙的市井里忙碌的穿梭，