FZU 1686 dlx重复覆盖

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <queue>
#include <climits>
#include <cmath>

using namespace std;
#define N 230
#define MAXNODE 60000
const int INF = INT_MAX;
const double eps = 1e-8;

int n,m,n1,m1;

struct DLX{
    int n,m,size;
    int U[MAXNODE] , D[MAXNODE] , L[MAXNODE] , R[MAXNODE];
    int col[MAXNODE] , row[MAXNODE];
    int first[N] , cnt_col[N] , ans;
    bool v[MAXNODE];

    void init(int _n , int _m)
    {
        n = _n , m = _m , size = _m;
        for(int i=0 ; i<=m ; i++){
            U[i] = D[i] = i;
            L[i] = i-1 , R[i] = i+1;
            col[i] = i , row[i] = 0;
        }
        L[0] = m , R[m] = 0;
        for(int i=1 ; i<=n ; i++) first[i] = -1;
        for(int i=1 ; i<=m ; i++) cnt_col[i] = 0;
        ans = INF;
    }

    void link(int r , int c)
    {
        ++size;
        U[D[c]] = size , D[size] = D[c];
        U[size] = c , D[c] = size;
        cnt_col[c]++;

        if(first[r]<0) first[r] = L[size] = R[size] = size;
        else{
            R[size] = R[first[r]] , L[R[first[r]]] = size;
            L[size] = first[r] , R[first[r]] = size;
        }
        row[size] = r , col[size] = c;
    }

    void Remove(int c)
    {
        for(int i=D[c] ; i!=c ; i=D[i]){
            L[R[i]] = L[i] , R[L[i]] = R[i];
        }
    }

    void Resume(int c)
    {
        for(int i=U[c] ; i!=c ; i=U[i]){
            L[R[i]] = R[L[i]] = i;
        }
    }

    int f()
    {
        int ret = 0;
        for(int c=R[0] ; c!=0 ; c=R[c]) v[c]=true;
        for(int c=R[0] ; c!=0 ; c=R[c])
            if(v[c]){
                ret++;
                v[c] = false;
                for(int i=D[c] ; i!=c ; i=D[i])
                    for(int j=R[i] ; j!= i ; j=R[j])
                        v[col[j]]=false;
            }
        return ret;
    }

    void Dance(int d)
    {
        if(d+f() >= ans) return;
        if(!R[0]){
            ans = min(ans , d);
            return;
        }
        int st = R[0];
        for(int i=R[0] ; i!=0 ; i=R[i])
            if(cnt_col[st]>cnt_col[i]) st=i;

        for(int i=D[st] ; i!=st ; i=D[i]){
            Remove(i);
            for(int j=R[i] ; j!=i ; j=R[j]) Remove(j);
            Dance(d+1);
            for(int j=L[i] ; j!=i ; j=L[j]) Resume(j);
            Resume(i);
        }
        return;
    }
}dlx;

int mat[N][N] , id[N][N];

int main()
{
  //  freopen("a.in" , "r" , stdin);
    while(~scanf("%d%d" , &n , &m))
    {
        int size = 0;
        for(int i=0 ; i<n ; i++)
            for(int j=0 ; j<m ; j++)
            {
                scanf("%d" , &mat[i][j]);
                //这里可以适当加优化，只保留那些有怪兽存在的位置
                if(mat[i][j]) size++ , id[i][j] = size;
            }
        scanf("%d%d" , &n1 , &m1);
        int t = (n-n1+1)*(m-m1+1);

        dlx.init(t , size);
        for(int k=0 ; k<t ; k++){
            int row = k/(m-m1+1);
            int col = k%(m-m1+1);
            for(int i=row ; i<row+n1 ; i++)
                for(int j=col ; j<col+m1 ; j++)
                    if(mat[i][j]) dlx.link(k+1 , id[i][j]);
        }
        dlx.Dance(0);
        printf("%d
" , dlx.ans);
    }
    return 0;
}