//题目意思:给出一个有向图,给出起点和终点,一个人从起点到终点去约会,
//每次走的路程要求是最小。没条路不能重复走,问可以走多少次。
//思路:如果没限制直接就可以跑最大流了,
//加了限制后就是说只有最短路径的边才能走,先求出起点到所有点的最短距离,
//然后找出每条边的起点的距离加上边的权值=终点的距离,那么这条边就可以走
#include <map>
#include <set>
#include <stack>
#include <queue>
#include <cmath>
#include <ctime>
#include <vector>
#include <cstdio>
#include <cctype>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
using namespace std;
#define INF 0x3f3f3f3f
#define inf -0x3f3f3f3f
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define mem0(a) memset(a,0,sizeof(a))
#define mem1(a) memset(a,-1,sizeof(a))
#define mem(a, b) memset(a, b, sizeof(a))
typedef long long ll;
const int maxn=1010;
struct Edge{
int from,to,flow,cap;
Edge(int u,int v,int w,int f):
from(u),to(v),cap(w),flow(f){}
};
struct Dinic{
int n,m,s,t; //结点数,边数(包括反相弧),源点编号和汇点编号
vector<Edge>edges; //边表,edges[e]和edges[e^1]互为反相弧
vector<int>G[maxn]; //邻接表,G[i][j]表示结点i的第j条边在e数组中的序号
bool vis[maxn]; //BFS使用
int d[maxn]; //从起点到i的距离
int cur[maxn]; //当前弧下标
void AddEdge(int from,int to,int cap){
edges.push_back(Edge(from,to,cap,0));
edges.push_back(Edge(to,from,0,0));
m=edges.size();
G[from].push_back(m-2);
G[to].push_back(m-1);
}
void init(int n){
this->n=n;
for(int i=0;i<=n;i++)
G[i].clear();
edges.clear();
}
bool BFS(){
mem0(vis);
queue<int>Q;
Q.push(s);
d[s]=0;
vis[s]=1;
while(!Q.empty()){
int x=Q.front();
Q.pop();
for(int i=0;i<G[x].size();i++){ //只考虑残量网络中的弧
Edge& e=edges[G[x][i]];
if(!vis[e.to]&&e.cap>e.flow){
vis[e.to]=1;
d[e.to]=d[x]+1;
Q.push(e.to);
}
}
}
return vis[t];
}
ll DFS(int x,int a){
if(x==t||a==0)
return a;
ll flow=0,f;
for(int& i=cur[x];i<G[x].size();i++){
Edge& e=edges[G[x][i]];
if(d[x]+1==d[e.to]&&(f=DFS(e.to,min(a,e.cap-e.flow)))>0){
e.flow+=f;
edges[G[x][i]^1].flow-=f;
flow+=f;
a-=f;
if(a==0)
break;
}
}
return flow;
}
ll Maxflow(int s,int t){
this->s=s;
this->t=t;
ll flow=0;
while(BFS()){
mem0(cur);
flow+=DFS(s,INF);
}
return flow;
}
};
struct Edge1{
int from,to,len;
Edge1(int u,int v,int c):
from(u),to(v),len(c){}
};
vector<Edge1>edges;
vector<int>G[maxn];
int dist[maxn];
int vis[maxn];
void init(int n){
for(int i=0;i<=n;i++)
G[i].clear();
edges.clear();
}
void Addedge(int u,int v,int c){
edges.push_back(Edge1(u,v,c));
int m=edges.size();
G[u].push_back(m-1);
}
void spfa(int s){
memset(dist,0x3f,sizeof(dist));
mem0(vis);
queue<int>Q;
Q.push(s);
dist[s]=0;
vis[s]=1;
while(!Q.empty()){
int x=Q.front();
Q.pop();
vis[x]=0; //该开始忘记了松弛操作
for(int i=0;i<G[x].size();i++){
Edge1& e=edges[G[x][i]];
if(dist[e.to]>dist[x]+e.len){
dist[e.to]=dist[x]+e.len;
if(!vis[e.to]){
vis[e.to]=1;
Q.push(e.to);
}
}
}
}
}
Dinic ZYC;
int main(){
int t;
int n,m;
scanf("%d",&t);
while(t--){
scanf("%d%d",&n,&m);
init(n);
int x,y,z;
for(int i=0;i<m;i++){
scanf("%d%d%d",&x,&y,&z);
if(x!=y)
Addedge(x,y,z);
}
int start,end;
scanf("%d%d",&start,&end);
spfa(start);
ZYC.init(n);
for(int i=1;i<=n;i++){
for(int j=0;j<G[i].size();j++){
Edge1& e=edges[G[i][j]];
if(dist[e.to]==dist[i]+e.len){
ZYC.AddEdge(i,e.to,1);
}
}
}
printf("%I64d\n",ZYC.Maxflow(start,end));
}
return 0;
}
3416 Marriage Match IV
最新推荐文章于 2020-03-02 11:14:43 发布