[DLX] hust 1017 Exact cover

来源：互联网发布：手机淘宝购物车闪退编辑：程序博客网时间：2024/05/20 01:09
题意：
给你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;}
0 0