Dancing Links 学习笔记

Dancing Links

本周的AI引论作业布置了一道数独

加了奇怪剪枝仍然TLE的Candy?不得不去学了dlx

dlxnb!

Exact cover

设全集X,X的若干子集的集合为S。精确覆盖是指,选择一个S的子集S‘,满足X中的每一个元素在S’中恰好出现一次。

是一个NPC问题。

可以表示成01矩阵形式,选择若干行,使得每一列恰好有且仅有一行为1.

Sudoku

数独可以转化为精确覆盖问题。

令N=81为数独中格子个数,则:

  1. (x, y)=1表示(x,y)处填了数
  2. (x+N, z)=1表示x行填了z
  3. (y+N*2, z)=1表示y列填了z
  4. (r+N*3, z)=1表示r宫填了z

对于已经填了数的格子,转化为1行;

对于空的格子,转化为9行。

Algorithm X

一种显然的dfs:

  • 就是选择某一列,再选择该列的为1的某一行。
  • 删除该列(包括该列上为1的所有行)
  • 删除该行(包括该行上为1的所有列)

一个显然的启发式优化:minimum-remaining-values(MRV) heuristic

  • 优先选择节点个数(1的个数)少的列。

Dancing links is the technique suggested by Donald Knuth to efficiently implement his Algorithm X.

是一种用来高效实现algorithm X的数据结构。

就是“交叉十字循环双向链表”。

第0行分别是root和每一列的列首节点

其他的只有为1的位置才有节点。

删除某一列时,只要处理该列首节点(包括其左右节点)的左右指针;

删除某列时同时要删除该列上为1的所有行;

删除某一行时,只要处理该行所有节点(包括其上下节点)的上下指针。

值得注意的是,删除之后该列/行的结构没有改变。

dancing links

实现细节

每个节点维护:

  • l r u d 左右上指针
  • col 列指针
  • row 行标号
  • cnt 保存该列的元素个数(只列首/用来MRV优化)

ah数组保存列首/行首节点指针

初始化init

  • 处理列首
  • 加在a[c]下,h[r]
  • (实际的“线”是不是直的不重要

删除某列del

  • 删除该列,以及该列上的所有行

恢复某列add

  • 按删除相反的顺序恢复

主过程dance

  1. root->r == root时完成

  2. 选择元素最少的某列c并删除该列(包括该列上为1的所有行)

  3. 选择该列上为1的某行,删除该行(包括该行上为1的所有列)

    实际上这一行在2中已经删除了,只要处理该行的列即可

  4. 递归搜索

  5. 恢复该行

  6. 恢复该列

==注意==

  1. del/add时处理个数是必要的,因为那一行所对应的列不一定会被删去
  2. 恢复要按照删除的逆序

代码

POJ 3076 16*16数独问题的代码

结构体版太丑了还是放指针版吧QwQ

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <ctime>
using namespace std;
const int NUM = 260*4*16, N = 260*16, K = 16, L = 4, M = N*16;

int n = 256*4, num = 256, m=0;
struct meow {
    meow *l, *r, *u, *d, *col;
    int row;
    int cnt;
} pool[NUM];
meow *a[NUM], *h[M], *root;
int ans[N], sz;
char s[20][20];
struct action {
    int x, y, z;
} q[M];

void init() {
    for(int i=0; i<=n; i++) a[i] = &pool[i];
    for(int i=0; i<=n; i++) {
        a[i]->l = a[i-1];
        a[i]->r = a[i+1];
        a[i]->u = a[i]->d = a[i];
        a[i]->col = a[i];
        a[i]->row = 0;
        a[i]->cnt = 0;
    }
    a[0]->l = a[n]; a[n]->r = a[0];
    root = a[0];
    sz = n;
    memset(h, 0, sizeof(h));
}
void link(int r, int c) {
    sz++;
    meow *x = a[sz] = &pool[sz];
    x->row = r;
    x->col = a[c];
    a[c]->cnt++;
    x->d = a[c]->d; x->d->u = x;
    x->u = a[c]; x->u->d = x;
    if(h[r] == NULL) {
        h[r] = x->l = x->r = x;
    }
    else {
        x->r = h[r]->r; x->r->l = x;
        x->l = h[r]; x->l->r = x;
    }
}
void del(meow *x) {
    x->l->r = x->r;
    x->r->l = x->l;
    for(meow *i = x->d; i != x; i = i->d)
        for(meow *j = i->r; j != i; j = j->r) {
            j->d->u = j->u;
            j->u->d = j->d;
            j->col->cnt--;
        }
}
void add(meow *x) {
    x->l->r = x->r->l = x;
    for(meow *i = x->u; i != x; i = i->u)
        for(meow *j = i->l; j != i; j = j->l) {
            j->u->d = j->d->u = j;
            j->col->cnt++;
        }
}
bool dance(int k) {
    if(root->r == root) {
        for(int i=1; i<=num; i++) {
            action &x = q[ans[i]];
            s[x.x][x.y] = 'A' + x.z-1;
        }
        return true;
    }
    meow *c = root; c->cnt = 1e9;
    for(meow *x = root->r; x != root; x = x->r)
        if(x->cnt < c->cnt) c = x;
    del(c);
    for(meow *i = c->d; i != c; i = i->d) {
        ans[k+1] = i->row;
        for(meow *j = i->r; j != i; j = j->r) del(j->col);
        if(dance(k+1)) return true;
        for(meow *j = i->l; j != i; j = j->l) add(j->col);
    }
    add(c);
    return false;
}
inline int grid_id(int x, int y, int k=L) {return (x-1)/k*k + (y-1)/k+1;}
void sudoku(int x, int y, int z) {
    m++;
    link(m, (x-1)*K+y);
    link(m, (x-1)*K+z + num);
    link(m, (y-1)*K+z + num*2);
    link(m, (grid_id(x, y)-1)*K+z + num*3);
    q[m] = (action) {x, y, z};
}
int main() {
    while(scanf("%s", s[1]+1) != EOF) {
        init();
        for(int i=1; i<=K; i++) {
            for(int j=1; j<=K; j++) {
                int a;
                if(s[i][j] == '-') a = 0;
                else a = s[i][j]-'A'+1;
                if(a != 0) sudoku(i, j, a);
                else for(int k=1; k<=K; k++) sudoku(i, j, k);
            }
            if(i != K) scanf("%s", s[i+1]+1);
        }
        dance(0);
        for(int i=1; i<=K; i++) {
            for(int j=1; j<=K; j++) printf("%c", s[i][j]);
            puts("");
        }
        puts("");
    }
}
posted @ 2019-03-19 16:37 Candy? 阅读(...) 评论(...) 编辑 收藏