[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; }