pku 1459 Power Network EK&&Dinic算法解决。

http://poj.org/problem?id=1459

才开始套自己的那个模板就是过不了一直TLE,最后看了别人过得修改了一下自己的模板最后1391s险过。。。

本题是一个多源点，多汇点问题，建立超级源点与汇点，然后套用模板求解。。

EK O(V*E^2)

View Code

#include <cstdio>
#include <cstring>
#include <iostream>
#include <queue>
#define maxn 107
#define inf 99999999
using namespace std;
int g[maxn][maxn],level[maxn];
bool vt[maxn];
//分层
bool layer(int s,int e)
{
queue<int>q;
memset(level,-1,sizeof(level));
q.push(s);
level[s] = 1;
int i,p;
while (!q.empty())
{
p = q.front();
q.pop();
for (i = 0; i <= e; ++i)
{
if (g[p][i] > 0 && level[i] == -1)//满足还没有分层，且p与i有流量
{
level[i] = level[p] + 1;
if (i == e) { return true;}
else q.push(i);
}
}
}
return false;
}
int dinic(int s,int e)
{
deque<int>q;//利用双端队列
int i,p,vs,ve,min,pos,sum;
sum = 0;
while (layer(s,e))
{
memset(vt,false,sizeof(vt));
q.push_back(s);//从后端插入
vt[s] = true;
while (!q.empty())
{
p = q.back();//从后端取值
if (p == e)//如果得到了终点
{
min = inf;
for (i = 1; i < q.size(); ++i)
{
vs = q[i - 1]; ve = q[i];
if (g[vs][ve] > 0 && g[vs][ve] < min)
{
min = g[vs][ve];
pos = vs;
}
}
sum += min;
for (i = 1; i < q.size(); ++i)
{
vs = q[i - 1]; ve = q[i];
if (g[vs][ve] > 0)
{
g[vs][ve] -= min;
g[ve][vs] += min;
}
}
while (!q.empty() && q.back() != pos)//返回最小流量的边的定点，重新找增广路
{
vt[q.back()] = false;
q.pop_back();
}
}
else//继续找下一层
{
for (i = 0; i <= e; ++i)
{
if (g[p][i] > 0 && !vt[i] && level[i] == level[p] + 1)
{
vt[i] = true;
q.push_back(i);
break;
}
}
if (i > e) q.pop_back();
}
}
}
return sum;
}
int main()
{
int i,n,m,np,nc;
int x,y,v;
while (~scanf("%d%d%d%d",&n,&np,&nc,&m))
{
memset(g,0,sizeof(g));
for (i = 0; i < m; ++i)
{
scanf(" (%d,%d)%d",&x,&y,&v);
g[x][y] = v;
}
for (i = 0; i < np; ++i)
{
scanf(" (%d)%d",&x,&v);
g
[x] = v;
}
for (i = 0; i < nc; ++i)
{
scanf(" (%d)%d",&x,&v);
g[x][n + 1] = v;
}
printf("%d\n",dinic(n,n+1));
}
return 0;
}