1. 程式人生 > >Luogu P4205 [NOI2005]智慧珠遊戲

Luogu P4205 [NOI2005]智慧珠遊戲

國內少有的可以練習神仙演算法——DLX的好題怎麼可以被爆搜埋沒呢?

看到這題沒有DLX的題解所以寫一篇,不過貌似我實現的太弱(構圖太慢)所以速度上不是很快。

下面開始講題,但請保證你要先學會DLX。(dalao寫的超詳細DLX

首先仔細閱讀一遍題目,我們可以大致整理出題意:\(12\)塊拼圖填滿一個\(10\)行的三角形圖案

稍微分析下性質發現我們其實只要滿足:每個位置都有拼圖每種拼圖都出現了\(1\)即可。

所以類似於數獨我們可以把它轉化為一個精確覆蓋問題

行表示任意一種顏色的珠子不同放置(任意一個位置不同都算作不同)

\(55+12=67\)列。前\(55\)列表示每一個位置都要有珠子,後\(12\)

列要求每種顏色的珠子都要有。

考慮建圖,由於資料範圍實在很小,我們可以直接暴力搞。

對於每一種顏色,先手動打表求出每一個珠子關於一個珠子的相對座標差值。

然後直接列舉位置?但是這樣並沒有包括旋轉翻轉的情況,難道還要打一個\(8\)倍的表?

沒必要,我們可以直接列舉座標優先加到橫座標還是縱座標,同時列舉一下每次的正負號即可(這個具體看程式碼吧,很好理解的)

所以這個問題就被我們轉化為了一個\(2730\)行,\(67\)列,\(1\)的個數為\(15084\)的精確覆蓋問題了。直接上DLX即可。

值得一提的是作為正解的DLX碼量遠遠小於爆搜。(我寫了加上註釋\(106\)行,而且大括號都換行了)

CODE

#include<cstdio>
#include<cstring>
#define RI register int
#define CI const int&
#define Ms(f,x) memset(f,x,sizeof(f))
using namespace std;
const int N=15,multag[2]={-1,1};
const int length[12]={3,4,4,4,5,5,5,5,5,5,5,5}; //length of shapes
const int table[12][5][2]= //direction of shapes
{
    {{0,0},{1,0},{0,1}}, //A
    {{0,0},{0,1},{0,2},{0,3}}, //B
    {{0,0},{1,0},{0,1},{0,2}}, //C
    {{0,0},{1,0},{0,1},{1,1}}, //D
    {{0,0},{1,0},{2,0},{2,1},{2,2}}, //E
    {{0,0},{0,1},{1,1},{0,2},{0,3}}, //F
    {{0,0},{1,0},{0,1},{0,2},{1,2}}, //G
    {{0,0},{1,0},{0,1},{1,1},{0,2}}, //H
    {{0,0},{0,1},{0,2},{1,2},{1,3}}, //I
    {{0,0},{-1,1},{0,1},{1,1},{0,2}}, //J
    {{0,0},{1,0},{1,1},{2,1},{2,2}}, //K
    {{0,0},{1,0},{0,1},{0,2},{0,3}}, //L
};
const int numx[56]={0,1,2,2,3,3,3,4,4,4,4,5,5,5,5,5,6,6,6,6,6,6,7,7,7,7,7,7,7,8,8,8,8,8,8,8,8,9,9,9,9,9,9,9,9,9,10,10,10,10,10,10,10,10,10,10};
const int numy[56]={0,1,1,2,1,2,3,1,2,3,4,1,2,3,4,5,1,2,3,4,5,6,1,2,3,4,5,6,7,1,2,3,4,5,6,7,8,1,2,3,4,5,6,7,8,9,1,2,3,4,5,6,7,8,9,10};
int tx[2],nx[2],num[N][N],id[2750],numrow,numcol; char puzzle[N][N]; bool vis[N];
class Dancing_Links_AlgorithmX
{
    private:
        static const int N=2750,M=80,ALL=15100;
        int n,m,tot,L[ALL],R[ALL],U[ALL],D[ALL],Row[ALL],Col[ALL],H[N],S[M];
        #define Dance(i,A,s) for (i=A[s];i!=s;i=A[i])
        inline void remove(CI c)
        {
            L[R[c]]=L[c]; R[L[c]]=R[c]; RI i,j;
            Dance(i,D,c) Dance(j,R,i)
            U[D[j]]=U[j],D[U[j]]=D[j],--S[Col[j]];
        }
        inline void resume(CI c)
        {
            RI i,j; Dance(i,U,c) Dance(j,L,i)
            ++S[Col[U[D[j]]=D[U[j]]=j]];
            L[R[c]]=R[L[c]]=c;
        }
    public:
        inline void init(CI N,CI M)
        {
            n=N; m=M; for (RI i=0;i<=m;++i)
            S[i]=0,U[i]=D[i]=i,L[i]=i-1,R[i]=i+1;
            R[m]=0; L[0]=m; tot=m; Ms(H,0);
        }
        inline void link(CI r,CI c)
        {
            ++S[Col[++tot]=c]; Row[tot]=r;
            D[tot]=D[c]; U[D[c]]=tot; U[tot]=c; D[c]=tot;
            if (!H[r]) H[r]=L[tot]=R[tot]=tot; else
            R[tot]=R[H[r]],L[R[H[r]]]=tot,L[tot]=H[r],R[H[r]]=tot;
        }
        inline bool DFS(void)
        {
            if (!R[0]) return 1; RI i,j; int now=R[0];
            Dance(i,R,0) if (S[i]<S[now]) now=i;
            remove(now); Dance(i,D,now)
            {
                if (Col[i]<=55) puzzle[numx[Col[i]]][numy[Col[i]]]=id[Row[i]]+'A';
                Dance(j,R,i) remove(Col[j]),(Col[j]<=55)&&(puzzle[numx[Col[j]]][numy[Col[j]]]=id[Row[j]]+'A');
                if (DFS()) return 1; Dance(j,L,i) resume(Col[j]);
            }
            resume(now); return 0;
        }
        #undef Dance
}DLX;
inline void init(void)
{
    RI i,j; for (i=1;i<=10;++i) for (j=1;j<=i;++j) if (puzzle[i][j]!='.') vis[puzzle[i][j]-'A']=1;
    for (i=1;i<=10;++i) for (j=1;j<=i;++j) num[i][j]=++numcol; DLX.init(2730,numcol+12);
}
int main()
{
    //freopen("CODE.in","r",stdin); freopen("CODE.out","w",stdout);
    RI i; for (i=1;i<=10;++i) scanf("%s",puzzle[i]+1);
    //build excat cover model
    init(); for (RI cases=0,mx;cases<12;++cases)
    for (++numcol,mx=0;mx<2;++mx) for (RI dx=0;dx<2;++dx) for (RI dy=0;dy<2;++dy)
    for (tx[0]=1;tx[0]<=10;++tx[0]) for (tx[1]=1;tx[1]<=tx[0];++tx[1])
    {
        RI k; bool flag=1; for (k=0;k<length[cases];++k)
        {
            nx[mx]=tx[mx]+multag[dx]*table[cases][k][0];
            nx[mx^1]=tx[mx^1]+multag[dy]*table[cases][k][1];
            if (vis[cases]) { if (puzzle[nx[0]][nx[1]]!=cases+'A') { flag=0; break; } }
            else if (puzzle[nx[0]][nx[1]]!='.') { flag=0; break; }
        }
        if (!flag) continue; id[++numrow]=cases; DLX.link(numrow,numcol);
        for (k=0;k<length[cases];++k)
        {
            nx[mx]=tx[mx]+multag[dx]*table[cases][k][0];
            nx[mx^1]=tx[mx^1]+multag[dy]*table[cases][k][1];
            DLX.link(numrow,num[nx[0]][nx[1]]);
        }
    }
    //use Dancing Links to find solution
    if (!DLX.DFS()) return puts("No solution"),0;
    for (i=1;i<=10;++i,putchar('\n')) for (RI j=1;j<=i;++j)
    putchar(puzzle[i][j]); return 0;
}