P4304 [TJOI2013]攻击装置 【网络流】

题目描述

给定一个01矩阵，其中你可以在0的位置放置攻击装置。 每一个攻击装置（x,y）都可以按照“日”字攻击其周围的8个位置（x-1,y-2）,(x-2,y-1),(x+1,y-2),(x+2,y-1),(x-1,y+2),(x-2,y+1),(x+1,y+2),(x+2,y+1)

求在装置互不攻击的情况下，最多可以放置多少个装置。

输入格式

第一行一个整数N，表示矩阵大小为N*N。

接下来N行每一行一个长度N的01串，表示矩阵。

输出格式

一个整数，表示在装置互不攻击的情况下最多可以放置多少个装置。

输入输出样例

输入 #1复制

3
010
000
100

输出 #1复制

4

说明/提示

30%数据N<=50

100%数据 N<=200

思路

　　方格取数问题加强版、骑士共存问题的魔改

　　做过任意一个就可以秒了

CODE

#include <bits/stdc++.h>

using namespace std;



const int maxn = 1e6 + 7;

const int inf = 0x3f3f3f3f;



template<class T>inline void read(T &res)

{

    char c;T flag=1;

    while((c=getchar())<'0'||c>'9')if(c=='-')flag=-1;res=c-'0';

    while((c=getchar())>='0'&&c<='9')res=res*10+c-'0';res*=flag;

}



struct edge{int from,to,cap,flow;};

int s, t, n, m;

struct isap

{

    int n,s,t,p[maxn],d[maxn],cur[maxn],num[maxn];

    bool vis[maxn];

    vector<int>g[maxn];

    vector<edge>edges;

    void init(int n,int s,int t) {

        this->n = n;

        this->s = s;

        this->t = t;

        for(int i = 1;i <= n;i++) g[i].clear();

        edges.clear();

    }

    void addegde(int from,int to,int cap) {

        edges.push_back((edge){from, to, cap, 0});

        edges.push_back((edge){to, from, 0, 0});

        int m = edges.size();

        g[from].push_back(m-2);

        g[to].push_back(m-1);

    }



    int augment() {///找增广路

        int x = t,a = inf;

        while(x!=s) {

            a = min(a, edges[p[x]].cap - edges[p[x]].flow);

            x = edges[p[x]].from;

        }

        x=t;

        while(x != s) {

            edges[p[x]].flow += a;

            edges[p[x]^1].flow = -a;

            x = edges[p[x]].from;

        }

        return a;

    }

    int maxflow() {///更新最大流

        int flow = 0;

        memset(num, 0, sizeof(num));

        memset(cur, 0, sizeof(cur));

        for(int i = 1; i <= n; i++) num[d[i]]++;

        int x = s;

        while(d[s] < n) {///最长的一条链上，最大的下标是nv-1，如果大于等于nv说明已断层

            if(x == t) {

                flow += augment();

                x = s;//回退

            }

            bool ok = 0;

            for(int i = cur[x]; i < g[x].size(); i++) {

                edge &e = edges[g[x][i]];

                if(d[x] == d[e.to] + 1 && e.cap > e.flow) {

                    p[e.to] = g[x][i];

                    cur[x] = i;x = e.to;

                    ok = 1;

                    break;

                }

            }

            if(!ok) {

                int m = n-1;

                for(int i = 0; i < g[x].size();i++) {

                    edge &e=edges[g[x][i]];

                    if(e.cap>e.flow) m=min(m,d[e.to]);

                }

                num[d[x]]--;

                if(!num[d[x]]) break;

                d[x] = m+1;

                num[d[x]]++;

                cur[x] = 0;

                if(x != s) x = edges[p[x]].from;

            }

        }

        return flow;

    }

}ISAP;

inline int id(int x, int y) {
    return (x-1) * n + y;
}

char a[507][507];

int dr[8] = {-2, -2, -1, -1, 1, 1, 2, 2};
int dc[8] = {1, -1, 2, -2, 2, -2, 1, -1};

int main()
{
    read(n);
    
    memset(a, 0, sizeof(a));
    s = 0, t = n * n + 1;
    ISAP.init(n * n * 2, s, t);
    for ( int i = 1; i <= n; ++i ) {
        for ( int j = 1; j <= n; ++j ) {
            cin >> a[i][j];
            if(a[i][j] == '1')
                ++m;
        }
    }
    // for ( int i = 1; i <= n; ++i ) {
    //     for ( int j = 1; j <= n; ++j ) {
    //         cout << a[i][j];
    //     }
    //     cout << endl;
    // }
    for ( int i = 1; i <= n; ++i ) {
        for ( int j = 1; j <= n; ++j ) {
            if(a[i][j] == '1')
                continue;
            if((i + j) & 1) {
                ISAP.addegde(s, id(i, j), 1);
                for ( int k = 0; k <= 7; ++k ) {
                    int x = i + dr[k], y = j + dc[k];
                    if(x < 1 || x > n || y < 1 || y > n) {
                        continue;
                    }
                    ISAP.addegde(id(i, j), id(x, y), 1);
                }
            }
            else {
                ISAP.addegde(id(i, j), t, 1);
            }
        }
    }
    int maxflow = ISAP.maxflow();
    //dbg(maxflow);
    int ans = n * n - m - maxflow;
    cout << ans << endl;
    return 0;
}