hdu 3277(最大流+二分+并查集）

来源：互联网发布：外交官新秀丽知乎编辑：程序博客网时间：2024/05/21 15:43
题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=3277
思路:3081差不多，就是多了每个女孩可以另外选任意k个不喜欢的,将每个女孩u分为u1，u2，若u喜欢v则加一条u1到v的边否则加一条u2到v的边，令加u1到u2的容量为k的边；
然后不知道为什么，wa了一下午，哎。。。
以下是wa代码，不知道哪儿有bug.
View Code
  1 #include<iostream>  2 #include<cstdio>  3 #include<cstring>  4 using namespace std;  5 #define MAXM 22222222  6 #define MAXN 888  7 #define inf 1<<30  8 struct Edge{  9     int v,cap,next; 10 }edge[MAXM],E[MAXM]; 11  12 int map[MAXN][MAXN]; 13 int cur[MAXN]; 14 int head[MAXN]; 15 int pre[MAXN]; 16 int level[MAXN]; 17 int gap[MAXN]; 18 int NV,NE,n,m,k,f; 19 int parent[MAXN]; 20 int girl[MAXN],boy[MAXN]; 21  22  23 void Initiate(){ 24     for(int i=1;i<=n;i++){ 25         parent[i]=-1; 26     } 27 } 28  29  30 int Find(int x){ 31     int s; 32     for(s=x;parent[s]>0;s=parent[s]) 33         ; 34     while(x!=s){ 35         int tmp=parent[x]; 36         parent[x]=s; 37         x=tmp; 38     } 39     return s; 40 } 41  42 void Union(int u,int v){ 43     int r1=Find(u); 44     int r2=Find(v); 45     if(r1==r2)return ; 46     if(parent[r1]>parent[r2]){ 47         parent[r2]+=parent[r1]; 48         parent[r1]=r2; 49     }else { 50         parent[r1]+=parent[r2]; 51         parent[r2]=r1; 52     } 53 } 54  55  56 /* 57 int Find(int x){ 58     return parent[x]==x?x:(parent[x]=Find(parent[x])); 59 } 60  61 void Union(int u,int v){ 62     int r1=Find(u); 63     int r2=Find(v); 64     if(r1!=r2){ 65         r1<r2?parent[r2]=r1:parent[r1]=r2; 66     } 67 } 68 */ 69  70  71 int SAP(int vs,int vt){ 72     memset(pre,-1,sizeof(pre)); 73     memset(level,0,sizeof(level)); 74     memset(gap,0,sizeof(gap)); 75     for(int i=0;i<=3*n+1;i++)cur[i]=head[i]; 76     int u=pre[vs]=vs,maxflow=0,aug=-1; 77     gap[0]=NV; 78     while(level[vs]<NV){ 79 loop: 80         for(int &i=cur[u];i!=-1;i=edge[i].next){ 81             int v=edge[i].v; 82             if(edge[i].cap&&level[u]==level[v]+1){ 83                 aug==-1?aug=edge[i].cap:(aug=min(aug,edge[i].cap)); 84                 pre[v]=u; 85                 u=v; 86                 if(v==vt){ 87                     maxflow+=aug; 88                     for(u=pre[u];v!=vs;v=u,u=pre[u]){ 89                         edge[cur[u]].cap-=aug; 90                         edge[cur[u]^1].cap+=aug; 91                     } 92                     aug=-1; 93                 } 94                 goto loop; 95             } 96         } 97         int minlevel=NV; 98         for(int i=head[u];i!=-1;i=edge[i].next){ 99             int v=edge[i].v;100             if(edge[i].cap&&minlevel>level[v]){101                 cur[u]=i;102                 minlevel=level[v];103             }104         }105         gap[level[u]]--;106         if(gap[level[u]]==0)break;107         level[u]=minlevel+1;108         gap[level[u]]++;109         u=pre[u];110     }111     return maxflow;112 }113 114 void Insert(int u,int v,int cap,int cc=0){115     edge[NE].v=v;edge[NE].cap=cap;116     edge[NE].next=head[u];head[u]=NE++;117 118     edge[NE].v=u;edge[NE].cap=cc;119     edge[NE].next=head[v];head[v]=NE++;120 }121 122 void Build(int cap){123     for(int i=0;i<NE;i++){124         edge[i]=E[i];125     }126     for(int i=0;i<4*n;i++){127         edge[i].cap=cap;128     }129 }130 131 132 int Binary_Search(){133     int low=0,high=n,tmp=0;134     NV=3*n+2;135     while(low<=high){136         int mid=(low+high)>>1;137         Build(mid);138         if(mid*n==SAP(0,3*n+1)){139             tmp=mid;140             low=mid+1;141         }else142             high=mid-1;143     }144     return tmp;145 }146 147 148 int main(){149     int _case,g,b,g1,g2;150     scanf("%d",&_case);151     while(_case--){152         scanf("%d%d%d%d",&n,&m,&k,&f);153         Initiate();154         //for(int i=1;i<=n;i++)parent[i]=i;155         memset(map,0,sizeof(map));156         memset(head,-1,sizeof(head));157         for(int i=1;i<=m;i++){158             scanf("%d%d",&girl[i],&boy[i]);159         }160         for(int i=1;i<=f;i++){161             scanf("%d%d",&g1,&g2);162             Union(g1,g2);163         }164         for(int i=1;i<=m;i++){map[Find(girl[i])][boy[i]]=1;}165         for(int i=1;i<=n;i++){166             for(int j=1;j<=n;j++){167                 if(map[Find(i)][j]==1)map[i][j]=1;168             }169         }170         NE=0;171         for(int i=1;i<=n;i++){172             Insert(0,i,1);//源点与女孩之间添边173         }174         for(int i=2*n+1;i<=3*n;i++){175             Insert(i,3*n+1,1);//男孩与汇点之间连边176         }177         //将每个女孩拆成i,i+n;178         for(int i=1;i<=n;i++){179             Insert(i,i+n,k);//女孩与女孩之间加边180         }181         for(int i=1;i<=n;i++){182             for(int j=1;j<=n;j++){183                 if(map[i][j]){    184                     Insert(i,j+2*n,1);//如果女孩和男孩连边，则i与男孩连边185                 }else {186                     Insert(i+n,j+2*n,1);//没有边，则i+n与男孩连边187                 }188             }189         }190         for(int i=0;i<NE;i++)E[i]=edge[i];191         int ans=Binary_Search();192         printf("%d\n",ans);193     }194     return 0;195 }196 197 198 199 200 201
0 0