题目链接: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 }else 142 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