POJ 2455Secret Milking Machine(二分+网络流之最大流)

时间:2023-03-09 19:37:50
POJ 2455Secret Milking Machine(二分+网络流之最大流)

题目地址:POJ2455

手残真浪费时间啊。。又拖到了今天才找出了错误。。每晚两道题不知不觉又变回了每晚一道题、。。sad。。

第一次在isap中忘记调用bfs,第二次则是遍历的时候竟然是从1開始遍历的。。。sad。。。

这题思路倒是非常easy,就是有一个比較坑的地方,就是这里的重边要当两条边来用,曾经受最短路什么的影响,直接把慢的删了,非常明显不正确。。。这里的两条重边都是能够走的。

建图思路是,将1当作源点,将n当作汇点。右边的地方就连边,注意是无向边。最后推断最大流是否等于道路条数。二分找最小值。

代码例如以下:

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <math.h>
#include <ctype.h>
#include <queue>
#include <map>
#include <algorithm> using namespace std;
const int INF=1e9;
int head[500], s, t, nv, cnt, sum;
int num[500], d[500], pre[500], cur[500], q[500];
struct node
{
int u, v, cap, next;
}edge[5000000];
struct N
{
int u, v, w;
}bian[5000000];
void add(int u, int v, int cap)
{
edge[cnt].v=v;
edge[cnt].cap=cap;
edge[cnt].next=head[u];
head[u]=cnt++; edge[cnt].v=u;
edge[cnt].cap=cap;
edge[cnt].next=head[v];
head[v]=cnt++;
}
void bfs()
{
memset(num,0,sizeof(num));
memset(d,-1,sizeof(d));
int f1=0, f2=0, i;
q[f1++]=t;
d[t]=0;
num[0]=1;
while(f1>=f2)
{
int u=q[f2++];
for(i=head[u];i!=-1;i=edge[i].next)
{
int v=edge[i].v;
if(d[v]==-1)
{
d[v]=d[u]+1;
num[d[v]]++;
q[f1++]=v;
}
}
}
}
int isap()
{
memcpy(cur,head,sizeof(cur));
int flow=0, u=pre[s]=s, i;
bfs();
while(d[s]<nv)
{
if(u==t)
{
int f=INF, pos;
for(i=s;i!=t;i=edge[cur[i]].v)
{
if(f>edge[cur[i]].cap)
{
f=edge[cur[i]].cap;
pos=i;
}
}
for(i=s;i!=t;i=edge[cur[i]].v)
{
edge[cur[i]].cap-=f;
edge[cur[i]^1].cap+=f;
}
flow+=f;
if(flow>=sum)
{
return flow;
}
u=pos;
}
for(i=cur[u];i!=-1;i=edge[i].next)
{
if(d[edge[i].v]+1==d[u]&&edge[i].cap)
{
break;
}
}
if(i!=-1)
{
cur[u]=i;
pre[edge[i].v]=u;
u=edge[i].v;
}
else
{
if(--num[d[u]]==0) break;
int mind=nv;
for(i=head[u];i!=-1;i=edge[i].next)
{
if(mind>d[edge[i].v]&&edge[i].cap)
{
mind=d[edge[i].v];
cur[u]=i;
}
}
d[u]=mind+1;
num[d[u]]++;
u=pre[u];
}
}
return flow;
}
int main()
{
int n, p, a, b, c, i, j, top;
while(scanf("%d%d%d",&n,&p,&sum)!=EOF)
{
top=0;
while(p--)
{
scanf("%d%d%d",&a,&b,&c);
bian[top].u=a;
bian[top].v=b;
bian[top++].w=c;
}
int high=1000001, mid, low=1, ans, x;
while(low<=high)
{
mid=(high+low)/2;
s=1;
t=n;
nv=t+1;
memset(head,-1,sizeof(head));
cnt=0;
for(i=0;i<top;i++)
{
if(bian[i].w<=mid)
{
add(bian[i].u,bian[i].v,1);
}
}
x=isap();
//printf("%d %d %d\n",x, mid,top);
if(x>=sum)
{
ans=mid;
high=mid-1;
}
else
{
low=mid+1;
}
}
printf("%d\n",ans);
}
return 0;
}