luogu3153&&bzoj1305 [CQOI2009]DANCE跳舞

http://www.elijahqi.win/2017/11/23/luogu3153bzoj1305-cqoi2009dance%e8%b7%b3%e8%88%9e/
Description
一次舞会有n个男孩和n个女孩。每首曲子开始时,所有男孩和女孩恰好配成n对跳交谊舞。每个男孩都不会和同一个女孩跳两首(或更多)舞曲。有一些男孩女孩相互喜欢,而其他相互不喜欢(不会“单向喜欢”)。每个男孩最多只愿意和k个不喜欢的女孩跳舞,而每个女孩也最多只愿意和k个不喜欢的男孩跳舞。给出每对男孩女孩是否相互喜欢的信息,舞会最多能有几首舞曲?
Input
第一行包含两个整数n和k。以下n行每行包含n个字符,其中第i行第j个字符为’Y’当且仅当男孩i和女孩j相互喜欢。
Output
仅一个数,即舞曲数目的最大值。
Sample Input
3 0
YYY
YYY
YYY
Sample Output
3
HINT
N<=50 K<=30
Source
加强数据By dwellings and liyizhen2
因为有不喜欢的人 所以我把男孩女孩拆点 分为喜欢和不喜欢两种点 然后男孩从喜欢向不喜欢建立k的边 女孩从不喜欢向喜欢建立k的边 然后我每次枚举每个男孩和女孩可以跳多少支舞 然后每次都把这个限制条件假如 源点到男孩的边 和女孩到汇点的边 判断能否满流

#include
#include
#include
#define N 55
#define inf 0x3f3f3f3f
#include
using namespace std;
inline int read(){
    int x=0;char ch=getchar();
    while (ch<'0'||ch>'9') ch=getchar();
    while (ch<='9'&&ch>='0'){x=x*10+ch-'0';ch=getchar();}
    return x;
}
struct node{
    int x,y,z,next; 
}data[16*N*N];
int num=1,c[N<<1],cnt,w[16*N*N],level[N<<2],h[N<<2],n,k,T;
char s1[N];
inline void insert1(int x,int y,int z){
    data[++num].y=y;data[num].z=z;data[num].next=h[x];h[x]=num;w[num]=z;data[num].x=x;
    data[++num].y=x;data[num].z=0;data[num].next=h[y];h[y]=num;w[num]=0;data[num].x=y;
}
inline int calc(int x,int y,int op){
    if (y==1) return (x-1)*2+op;else return n*2+(x-1)*2+op;
}
inline bool bfs(){
    memset(level,0,sizeof(level));level[0]=1;queue<int>q;q.push(0);
    while (!q.empty()){
        int x=q.front();q.pop();
        for (int i=h[x];i;i=data[i].next){
            int y=data[i].y,z=data[i].z;
            if (level[y]||!z) continue;q.push(y);level[y]=level[x]+1;if(y==T) return 1;
        }
    } return 0;
}
int dfs(int x,int s){
    if (x==T) return s;int ss=s;
    for (int i=h[x];i;i=data[i].next){
        int y=data[i].y,z=data[i].z;
        if (level[x]+1==level[y]&&z){
            int xx=dfs(y,min(z,s));
            s-=xx;data[i].z-=xx;data[i^1].z+=xx;
            if (!s) return ss;
        }
    }return ss-s;
}
inline bool check(int mid){
    int ans=0;for (int i=1;i<=cnt;++i) data[c[i]].z=mid;
    while (bfs()) ans+=dfs(0,inf);
    if (ans==mid*n) return 1;else return 0;
}
int main(){
    freopen("bzoj1305.in","r",stdin);
    n=read();k=read();
    for (int i=1;i<=n;++i){
        scanf("%s",s1+1);
        for (int j=1;j<=n;++j)
            if (s1[j]!='Y') insert1(calc(i,1,2),calc(j,2,2),1);
            else insert1(calc(i,1,1),calc(j,2,1),1);
    }T=4*n+1;
    for (int i=1;i<=n;++i) insert1(calc(i,1,1),calc(i,1,2),k);
    for (int i=1;i<=n;++i) insert1(calc(i,2,2),calc(i,2,1),k);
    for (int i=1;i<=n;++i) insert1(0,calc(i,1,1),inf),c[++cnt]=num-1;
    for (int i=1;i<=n;++i) insert1(calc(i,2,1),T,inf),c[++cnt]=num-1;
    //for (int i=2;i<=num;++i) printf("%d %d %d\n",data[i].x,data[i].y,data[i].z);
    int l=0,r=n;
    while (l<=r){
        int mid=l+r>>1;for (int i=2;i<=num;++i) data[i].z=w[i];
        if (check(mid)) l=mid+1;else r=mid-1;
    }printf("%d",r);
    return 0;
}

你可能感兴趣的:(网络流)