题目:http://acm.hdu.edu.cn/showproblem.php?pid=3416
题意:题目就是要多次从A点到B点,每次一定要选一条A到B的最短路,并且这条最短路上的边之前是没有走过的,点可以多次经过。问最多能进行多少次。
比较综合的一道题目,下面我提到的次数都是指题目要求的次数。
如果存在一条最短路是从A到C再到B,那么A经过C到B的次数,等于A到C的次数 和 C到B的次数的最小值。
这样分析的话,我们会发现这其实就是个网络流问题,我们要找出的就是哪些边是在最短路上的,每条边容量1,求个最大流即可。
我是用迪杰斯特拉算出A点到其它点的最短路径长度,接下来就是判断哪些边在最短路上,其实这个也好判断,记d[i]是A点到i点的最短距离,假设某条边X->Y,距离为C,如果d[X] + C = d[Y],就说明这条边在最短路上。
所以整个流程就是,求最短路,构造网络流的图,求最大流。
细心点码就没问题的,几个注意的地方,一个是这个图是有向图,另外就是可能存在重边和自环(自环可以忽略)。
PS:因为偷懒,所以里面有部分数组(V和d)是在求最短路和最大流都用到的,但是意义不一样。。。
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#include<queue>
using namespace std;
#define pb push_back
inline void in(int &x){
char c=getchar();
x = 0;
while(c<48 || c>57) c=getchar();
while(c>=48 && c<=57){
x = x*10+c-48;
c = getchar();
}
}
const int maxn = 1001;
const int maxm = 100001;
const int inf = 0x7fffffff;
struct Edge{
int from, to, cap, flow;
Edge(){}
Edge(int from, int to, int cap, int flow):from(from),to(to),cap(cap),flow(flow){}
};
vector<int> V[maxn];
vector<Edge> G;
int t, n, m, a[maxm], b[maxm], c[maxm];
struct Node{
int id, d;
Node(){}
Node(int id, int d):id(id),d(d){}
bool operator < (const Node& tmp)const{
return d > tmp.d;
}
};
void add(int from, int to, int cap){
G.pb(Edge(from, to, cap, 0));
G.pb(Edge(to, from, 0, 0));
int x = G.size();
V[from].pb(x-2);
V[to].pb(x-1);
}
bool done[maxn];
int st, tar, d[maxn];
int sz[maxn][maxn];
void dij(){
memset(done,0,sizeof(done));
for(int i=1; i<=n; i++) d[i]=inf;
priority_queue<Node> Q;
Q.push(Node(st, 0));
d[st] = 0;
while(!Q.empty()){
Node nd=Q.top(); Q.pop();
int x = nd.id;
if(done[x]) continue;
done[x] = 1;
for(int i=0; i<V[x].size(); i++){
int j = V[x][i];
if(done[b[j]]) continue;
int tmp = d[x] + c[j];
if(tmp < d[b[j]]){
d[b[j]] = tmp;
Q.push(Node(b[j], tmp));
}
}
}
}
void init_flow(){
memset(sz,0,sizeof(sz));//因为可能有重边,用它来合并,sz[i][j]同时也代表了i到j的容量
for(int i=0; i<m; i++){
if(a[i]!=b[i] && d[a[i]]+c[i] == d[b[i]]){
sz[a[i]][b[i]]++;
}
}
G.clear();
V[0].clear();
add(0, st, inf);
for(int i=1; i<=n; i++){
V[i].clear();
for(int j=1; j<=n; j++){
if(sz[i][j]) add(i, j, sz[i][j]);
}
}
}
bool bfs(){
memset(d,-1,sizeof(d));
queue<int> Q;
Q.push(st);
d[st]=0;
while(!Q.empty()){
int x=Q.front(); Q.pop();
for(int i=0; i<V[x].size(); i++){
Edge &e = G[V[x][i]];
if(d[e.to]==-1 && e.cap>e.flow){
d[e.to] = d[x]+1;
Q.push(e.to);
}
}
}
return d[tar]!=-1;
}
int cur[maxn];
int dfs(int x, int v){
if(x==tar || v==0) return v;
int flow=0, f;
for(int &i = cur[x]; i<V[x].size(); i++){
int j=V[x][i];
Edge &e = G[j];
if(d[e.to]==d[x]+1 && (f=dfs(e.to, min(v, e.cap-e.flow)))>0){
v-=f;
flow+=f;
e.flow+=f;
G[j^1].flow-=f;
if(!v) break;
}
}
return flow;
}
int max_flow(){
int flow=0;
while(bfs()){
memset(cur,0,sizeof(cur));
flow += dfs(st, inf);
}
return flow;
}
int main(){
#ifndef ONLINE_JUDGE
freopen("001.in", "r", stdin);
#endif
in(t);
while(t--){
in(n); in(m);
for(int i=1; i<=n; i++) V[i].clear();
for(int i=0; i<m; i++){
in(a[i]); in(b[i]); in(c[i]);
if(a[i]==b[i]) continue;
V[a[i]].pb(i);
}
in(st); in(tar);
dij();//迪杰斯特拉
init_flow();//构造网络流的图
printf("%d\n", max_flow());
}
return 0;
}