POJ 3740 Easy Finding (DLX)

题意:

精确覆盖,DLX模板题

思路:

这个博客写的不错
刚刚开学DLX,感觉是个很简单的东西,只能算是十字链表优化了DFS,所以理论上有些DLX题目的有些题目应该DFS也能水过去
去看板子,发现所有人的板子都几乎一模一样。。。

错误及反思:

代码:

#include <iostream>
#include <stdio.h>
#include <string.h>

const int MN = 1005;//最大行数
const int MM = 1005;//最大列数
const int MNODE = 1e5 + 5 + MM; //最大点数

struct DLX
{
    int n, m, si;
    //n行数m列数si目前有的节点数
    //十字链表组成部分
    int U[MNODE], D[MNODE], L[MNODE], R[MNODE];
    int Row[MNODE], Col[MNODE];
    //第i个结点的U向上指针D下L左R右,所在位置Row行Col列
    int H[MN], S[MM];
    //记录行的选择情况和列的覆盖情况
    int ansd, ans[MN];
    void init(int _n, int _m)  //初始化空表
    {
        n = _n;
        m = _m;
        for (int i = 0; i <= m; i++) //初始化第一横行(表头)
        {
            U[i] = D[i] = i;      //目前纵向的链是空的
            L[i] = i - 1;
            R[i] = i + 1;         //横向的连起来
        }
        R[m] = 0; L[0] = m;
        si = m;                 //目前用了前0~m个结点

        memset(S, 0, sizeof(S));
        memset(H, -1, sizeof(H));
    }
    void link(int r, int c)    //插入点(r,c)
    {
        ++S[Col[++si] = c];     //si++;Col[si]=c;S[c]++;
        Row[si] = r;//si该结点的行数为r
        D[si] = D[c];//向下指向c的下面的第一个结点
        U[D[c]] = si;//c的下面的第一个结点的上面为si
        U[si] = c;//si的上面为列指针
        D[c] = si;//列指针指向的第一个该列中的元素设为si
        if (H[r]<0)//如果第r行没有元素
            H[r] = L[si] = R[si] = si;
        else
        {
            R[si] = R[H[r]];//si的右边为行指针所指的右边第一个元素
            L[R[H[r]]] = si;//行指针所指的右边第一个元素的左侧为si
            L[si] = H[r];//si的左侧为行指针
            R[H[r]] = si;//行指针的右侧为si
        }
    }
    void remove(int c)        //列表中删掉c列
    {
        L[R[c]] = L[c];//表头操作  //c列头指针的右边的元素的左侧指向c列头指针左边的元素
        R[L[c]] = R[c];//c列头指针的左边的元素的右侧指向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)        //恢复c列
    {
        for (int i = U[c]; i != c; i = U[i])//枚举该列元素
            for (int j = L[i]; j != i; j = L[j])//枚举该列元素所在的行
                ++S[Col[U[D[j]] = D[U[j]] = j]];//D[U[j]]=j;U[D[j]]=j;S[Col[j]]++;
        L[R[c]] = R[L[c]] = c;//c列头指针左右相连
    }
    bool dance(int d) //选取了d行
    {
        if (R[0] == 0)//全部覆盖了
        {
            //全覆盖了之后的操作
            ansd = d;
            return 1;
        }
        int c = R[0];//表头结点指向的第一个列
        for (int i = R[0]; i != 0; i = R[i])//枚举列头指针
            if (S[i]<S[c])//找到列中元素个数最少的
                c = i;
        remove(c);//将该列删去
        for (int i = D[c]; i != c; i = D[i])
        {//枚举该列的元素
            ans[d] = Row[i];//记录该列元素的行
            for (int j = R[i]; j != i; j = R[j])
                remove(Col[j]);//将该列的某个元素的行上的元素所在的列都删去
            if (dance(d + 1))
                return 1;
            for (int j = L[i]; j != i; j = L[j])
                resume(Col[j]);
        }
        resume(c);
        return 0;
    }
}dlx;
inline int read() {
    char ch = getchar(); int x = 0, f = 1;
    while(ch < '0' || ch > '9') {
        if(ch == '-') f = -1;
        ch = getchar();
    } while('0' <= ch && ch <= '9') {
        x = x * 10 + ch - '0';
        ch = getchar();
    } return x * f;
}
int ma[MN][MM];
int main()
{
    int n, m;
    while (scanf("%d%d", &n, &m) != EOF)
    {
        dlx.init(n, m);
        for (int i = 1; i <= n; i++)
        {
            for(int j = 1;j<=m;j++){
                ma[i][j]=read();
                if(ma[i][j])
                    dlx.link(i,j);
            }
        }

        if (dlx.dance(0))
            printf("Yes, I found it\n");
        else
            printf("It is impossible\n");
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/roll_keyboard/article/details/80156309
今日推荐