hust1017 Exact cover DLX模板

     题意是一个N*M的01矩阵,选取若干行,使得选取的集合中没列恰好有一个1..精确覆盖的模板题,dancing links的知识可以去 http://blog.csdn.net/sunny606/article/details/7833551 看看,讲的还是挺明白的,或者直接去看Knuth 的论文<dancing links>。这类题模板基本不用怎么动,难点在建表,能把题意转化成精确覆盖的模型就没什么难度了....

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
typedef long long ll;
const int maxn=1500*1500;

struct DLX
{
    int col[maxn],row[maxn],ans[maxn];
    int S[maxn],H[maxn];
    int size,cnt;
    int U[maxn],D[maxn],L[maxn],R[maxn];//十字链表
    void remove(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[col[j]];
            }
        }
    }
    void resume(int c)
    {
        L[R[c]]=c;
        R[L[c]]=c;
        for (int i=U[c]; i!=c; i=U[i])
        {
            for (int j=R[i]; j!=i; j=R[j])
            {
                U[D[j]]=j;
                D[U[j]]=j;
                ++S[col[j]];
            }
        }
    }

    bool dance(int k)//dfs
    {
        int c=R[0];
        if (c==0)
        {
            cnt=k;
            return true;
        }
        for (int i=R[0]; i!=0; i=R[i])
        if (S[c]>S[i]) c=i;//启发搜索
        remove(c);
        for (int i=D[c]; i!=c; i=D[i])
        {
            ans[k]=row[i];
            for (int j=R[i]; j!=i; j=R[j]) remove(col[j]);
            if (dance(k+1)) return true;
            for (int j=L[i]; j!=i; j=L[j]) resume(col[j]);

        }
        resume(c);
        return false;
    }
    void init(int m)
    {
        for (int i=0; i<=m; i++)
        {
            S[i]=0;
            L[i]=i-1;
            R[i]=i+1;
            U[i]=D[i]=i;
        }
        L[0]=m;
        R[m]=0;
        size=m+1;
        memset(H,-1,sizeof H);
    }
    void link(int x,int y)//把矩阵中(x,y)位置的1插入到十字链表
    {
        ++S[col[size]=y];
        row[size]=x;
        D[size]=D[y];
        U[D[y]]=size;
        U[size]=y;
        D[y]=size;
        if (H[x]<0) H[x]=L[size]=R[size]=size;
        else
        {
            R[size]=R[H[x]];
            L[R[H[x]]]=size;
            L[size]=H[x];
            R[H[x]]=size;
        }
        size++;
    }
}dlx;
int n,m,k,num;
int main()
{
//    freopen("in.txt","r",stdin);
    while (~scanf("%d%d",&n,&m))
    {
        dlx.init(m);
        for (int i=1; i<=n; i++)
        {
            scanf("%d",&num);
            while (num--)
            {
                scanf("%d",&k);
                dlx.link(i,k);
            }
        }
        if (dlx.dance(0))
        {
            printf("%d",dlx.cnt);
            for (int i=0; i<dlx.cnt; i++)
            {
                printf(" %d",dlx.ans[i]);
            }
            printf("\n");
        }
        else
        puts("NO");
    }


    return 0;
}


你可能感兴趣的:(hust1017 Exact cover DLX模板)