HDU 3277 Marriage Match III

有N个男孩，N个女孩。每个女孩可以选择一个没有跟他吵过架的男孩结婚。如果女孩X和女孩Y是朋友，且Y没有和男孩Z吵过架，女孩X同样可以选择男孩Z和自己结婚。另外，如果A和B是朋友，B和C是朋友，那么A和C也必定是朋友。每一轮中，每个女孩还可以额外选择K个自己原本不打算和她配对的男孩配对。一旦所有的女孩都找到了男友，那么他们就可以开始一轮新的游戏了，在每一轮新的游戏中，他们将使用相同的规则进行游戏，但是所有的女孩都不会选择之前选择过的男友了。问他们最多可以进行几轮游戏。

跟HDU 3081基本一样，就是多了一个条件：每一轮中，每个女孩还可以额外选择K个自己原本不打算和她配对的男孩配对。

这里的处理是把每个女孩i拆成两个点，在这两个点之间连一条边(i,i',K),如果女孩i喜欢男孩j，就从i连一条边(i,j,1)，如果不喜欢，就从i'连一条边(i',j,1)，然后的处理跟原来的完全一样。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define INF 1<<30
#define maxn 1000
#define maxm 300000
using namespace std;

int v[maxm],next[maxm],w[maxm];
int first[maxn],d[maxn],work[maxn],q[maxn];
int e,S,T;
int n,m,K,f;
int like[256][256],map[256][256];
void init(){
    e = 0;
    memset(first,-1,sizeof(first));
}

void add_edge(int a,int b,int c){
    //printf("add:%d to %d,cap = %d
",a,b,c);
    v[e] = b;next[e] = first[a];w[e] = c;first[a] = e++;
    v[e] = a;next[e] = first[b];w[e] = 0;first[b] = e++;
}

int bfs(){
    int rear = 0;
    memset(d,-1,sizeof(d));
    d[S] = 0;q[rear++] = S;
    for(int i = 0;i < rear;i++){
        for(int j = first[q[i]];j != -1;j = next[j])
            if(w[j] && d[v[j]] == -1){
                d[v[j]] = d[q[i]] + 1;
                q[rear++] = v[j];
                if(v[j] == T)   return 1;
            }
    }
    return 0;
}

int dfs(int cur,int a){
    if(cur == T)    return a;
    for(int &i = work[cur];i != -1;i = next[i]){
        if(w[i] && d[v[i]] == d[cur] + 1)
            if(int t = dfs(v[i],min(a,w[i]))){
                w[i] -= t;w[i^1] += t;
                return t;
            }
    }
    return 0;
}

int dinic(){
    int ans = 0;
    while(bfs()){
        memcpy(work,first,sizeof(first));
        while(int t = dfs(S,INF))   ans += t;
    }
    return ans;
}

void floyd(){
    for(int k = 1;k <= n;k++)
    for(int i = 1;i <= n;i++)   if(map[i][k])
    for(int j = 1;j <= n;j++)   if(map[k][j])
    map[i][j] = 1;

    for(int k = 1;k <= n;k++)
    for(int i = 1;i <= n;i++)   if(map[i][k])
    for(int j = 1;j <= n;j++)   if(like[k][j])
    like[i][j] = 1;
}

bool test(int mid){
    init();
    for(int i = 1;i <= n;i++)
        add_edge(S,i,mid),add_edge(i,i+n,K);
    for(int i = 2*n+1;i <= 3*n;i++)
        add_edge(i,T,mid);
    for(int i = 1;i <= n;i++){
        for(int j = 1;j <= n;j++)
            if(like[i][j])  add_edge(i,j+2*n,1);
            else            add_edge(i+n,j+2*n,1);
    }
    int ans = dinic();
    //printf("ans = %d
",ans);
    if(ans == mid*n)    return true;
    return false;
}
int main()
{
    int kase;
    scanf("%d",&kase);
    while(kase--){
        scanf("%d%d%d%d",&n,&m,&K,&f);
        S = 0,T = 3*n+1;
        memset(map,0,sizeof(map));
        memset(like,0,sizeof(like));
        for(int i = 0;i < m;i++){
            int a,b;
            scanf("%d%d",&a,&b);
            like[a][b] = 1;
        }
        for(int i = 0;i < f;i++){
            int a,b;
            scanf("%d%d",&a,&b);
            map[a][b] = map[b][a] = 1;
        }
        floyd();
        int L = 0,R = n,ans = 0;
        while(L <= R){
            int mid = (L+R)>>1;
            if(test(mid)){
                ans = mid;
                L = mid+1;
            }
            else R = mid-1;
        }
        printf("%d
",ans);
    }
    return 0;
}