bzoj 4930: 棋盘

Description

给定一个n×n的棋盘，棋盘上每个位置要么为空要么为障碍。定义棋盘上两个位置(x,y),(u,v)能互相攻击当前仅

当满足以下两个条件：

1:x=u或y=v

2:对于(x,y)与(u,v)之间的所有位置，均不是障碍。

现在有q个询问，每个询问给定ki，要求从棋盘中选出ki个空位置来放棋子，问最少互相能攻击到的棋子对数是多少？

Input

第一行一个整数n。

接下来输入一个n×n的字符矩阵，一个位置若为.，则表示这是一个空位置，若为#，则为障碍。

第n+2行输入一个整数q代表询问个数。

接下来q行，每行一个整数k，代表要放的棋子个数。

n ≤ 50, q ≤ 10000, k ≤ 棋盘中空位置数量

Output

输出共q行，每行代表对应询问的最少的互相能攻击到的棋子对数。

不难得出费用流的模型，由于图的特殊性可用bfs求最小费用增广路

#include<cstdio>
#include<algorithm>
#include<queue>
const int inf=0x3f3f3f3f;
int n,q,id[][][],id1=,id2=,as[],ap=,d[],l[];
char s[][];
int es[],enx[],ev[],e0[],ep=,lt;
void ae(int a,int b){
    es[ep]=b;enx[ep]=e0[a];ev[ep]=;e0[a]=ep++;
    es[ep]=a;enx[ep]=e0[b];ev[ep]=;e0[b]=ep++;
}
bool cmp(int a,int b){
    return d[a]<d[b];
}
int q0[],ed[],tk=;
std::queue<int>q1;
void mins(int&a,int b){if(a>b)a=b;}
void cal(int w){
    if(w>id1)mins(lt,l[w]+d[w]);
    for(int i=e0[w];i;i=enx[i])if(ev[i]){
        int u=es[i];
        if(l[u]>l[w])l[u]=l[w],q1.push(u);
    }
}
int dfs(int w){
    if(ed[w]==tk)return ;
    ed[w]=tk;
    if(w>id1&&l[w]+d[w]==lt){
        ++d[w];
        return ;
    }
    for(int i=e0[w];i;i=enx[i])if(ev[i]&&l[es[i]]==l[w]&&dfs(es[i])){
        ev[i]=,ev[i^]=;
        return ;
    }
    return ;
}
bool cal(){
    lt=inf;
    for(int i=;i<=id1;++i)q0[i]=i,l[i]=d[i];
    for(int i=id1+;i<=id2;++i)l[i]=inf;
    std::sort(q0+,q0+id1+,cmp);
    int qp=;
    while(){
        if(qp<=id1){
            int w=q0[qp];
            if(l[w]!=d[w]){++qp;continue;}
            if(q1.size()&&l[q1.front()]<l[w])cal(q1.front()),q1.pop();
            else cal(w),++qp;
        }else if(q1.size())cal(q1.front()),q1.pop();
        else break;
    }
    int flow=;
    for(int i=;i<=id1;++tk,++i)if(l[i]==d[i]&&dfs(i))++flow,++d[i];
    for(int i=;i<=flow;++i)++ap,as[ap]=as[ap-]+lt;
    return flow;
}
int main(){
    scanf("%d",&n);
    for(int i=;i<=n;++i)scanf("%s",s[i]+);
    for(int i=;i<=n;++i){
        for(int j=;j<=n;++j){
            if(s[i][j]!='.')continue;
            id[i][j][]=s[i-][j]=='.'?id[i-][j][]:++id1;
            id[i][j][]=s[i][j-]=='.'?id[i][j-][]:++id2;
        }
    }
    for(int i=;i<=n;++i){
        for(int j=;j<=n;++j)if(s[i][j]=='.'){
            ae(id[i][j][],id1+id[i][j][]);
        }
    }
    id2+=id1;
    while(cal());
    for(scanf("%d",&q);q;--q){
        int x;
        scanf("%d",&x);
        printf("%d\n",as[x]);
    }
    return ;
}