bzoj1305: [CQOI2009]dance跳舞(網路流)

Hanks_o發表於2018-03-06

題目傳送門

解法:
挺水的最大流啊。
拆點一個控制連喜歡的一個控制連不喜歡的。
第一個點向第二個點連容量為k的邊表示只能選k個不喜歡的。。
然後建完圖我高興的以為又a了一題。結果wa了。
靜下來想想發現別人找到四個伴侶我可能只找到兩個。
這樣的話是不行的。
所以二分規定放多少首歌。
然後從st只流出這麼多的流量。如果滿流,說明這個歌的數量可以滿足。

程式碼實現:

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
using namespace std;
char map[110][110];
struct node {int x,y,c,next,other;}a[210000];int len,last[510];
void ins(int x,int y,int c) {
    int k1,k2;
    len++;k1=len;a[len].x=x;a[len].y=y;a[len].c=c;
    a[len].next=last[x];last[x]=len;
    len++;k2=len;a[len].x=y;a[len].y=x;a[len].c=0;
    a[len].next=last[y];last[y]=len;
    a[k1].other=k2;a[k2].other=k1;
}
int head,tail,list[510],st,ed,h[510];
bool bt_h() {
    memset(h,0,sizeof(h));h[st]=1;head=1;tail=2;list[1]=st;
    while(head!=tail) {
        int x=list[head];
        for(int k=last[x];k;k=a[k].next) {
            int y=a[k].y;
            if(h[y]==0&&a[k].c>0) {h[y]=h[x]+1;list[tail++]=y;}
        }head++;
    }if(h[ed]==0)return false;return true;
}
int find_flow(int x,int f) {
    if(x==ed)return f;
    int s=0,t;
    for(int k=last[x];k;k=a[k].next) {
        int y=a[k].y;
        if(h[y]==h[x]+1&&a[k].c>0&&s<f) {t=find_flow(y,min(a[k].c,f-s));s+=t;a[k].c-=t;a[a[k].other].c+=t;}
    }if(s==0)h[x]=0;return s;
}
int main() {
    int n,k;scanf("%d%d",&n,&k);st=4*n+1;ed=st+1;
    for(int i=1;i<=n;i++) scanf("%s",map[i]+1);
    int l=1,r=n,mid,ans=0;
    while(l<=r) {
        mid=(l+r)/2;len=0;memset(last,0,sizeof(last));
        for(int i=1;i<=n;i++) {ins(st,i,mid);ins(3*n+i,ed,mid);ins(i,i+n,k);ins(i+2*n,i+3*n,k);}
        for(int i=1;i<=n;i++)for(int j=1;j<=n;j++) {
            if(map[i][j]=='Y')ins(i,j+3*n,1);else ins(i+n,j+2*n,1);
        }int sum=0;while(bt_h()==true)sum+=find_flow(st,999999999);
        if(sum==mid*n) {l=mid+1;ans=mid;}else r=mid-1;
    }printf("%d\n",ans);
    return 0;
}