Marriage Match IV
题目链接:Marriage Match IV HDU - 3416
题意
给你一个图,问有多少条不同的最短路,有重边并且可当作分开的两边,其多条最短路之间不能有任何相同的边
思路
开始时,看到了每条边只能经过一次,我想到了费用流,可惜这里的数据不小,N为1000,M为1e5,按照费用流平均复杂度,F * E * log(v) 毫无疑问会T,那么可以怎么优化呢。
我们可以先跑一下最短路,找到找到所有可能组成最短路的边,这个判断就是 dis[to] == dis[now] + cost,如果满足,则一定是最短路中的关键组成。
既然有了这些边,那么我们再以这些边组成一个网络,跑一个最大流即可。
代码
#include <bits/stdc++.h>
using namespace std;
#define rep(i,j,k) for(int i = (int)j;i <= (int)k;i ++)
#define per(i,j,k) for(int i = (int)j;i >= (int)k;i --)
#define debug(x) cerr<<#x<<" = "<<(x)<<endl
#define mmm(a,b) memset(a,b,sizeof(a))
#define pb push_back
typedef double db;
typedef long long ll;
const int MAXN = (int)1e3+7;
const int INF = (int)0x3f3f3f3f;
// 用于表示表示边的结构体(终点、容量、反向边)
struct edge{
int to,cap,rev;
edge(int to = 0,int cap = 0,int rev = 0):to(to),cap(cap),rev(rev){}
};
int N,M;
vector<edge> G[MAXN];
int level[MAXN]; //顶点到源点的距离标号
int iter[MAXN]; //当前弧,在其之前的边已经没有用了
//向图中增加一条从s到t容量为cap的边
void add_edge(int from,int to,int cap) {
G[from].pb(edge(to,cap,G[to].size()));
G[to ].pb(edge(from,0,G[from].size()-1));
}
//通过BFS计算从源点出发的距离标号
void bfs(int s){
mmm(level,-1);
queue<int> qu;
level[s] = 0;
qu.push(s);
while (!qu.empty()) {
int v = qu.front(); qu.pop();
rep(i,0,G[v].size()-1) {
edge &e = G[v][i];
if (e.cap > 0 && level[e.to] < 0) {
level[e.to] = level[v] + 1;
qu.push(e.to);
}
}
}
}
//通过DFS寻找增广路
int dfs(int v,int t,int f) {
if (v == t) return f;
for (int &i = iter[v];i < G[v].size();i ++) {
edge &e = G[v][i];
if (e.cap > 0 && level[v] < level[e.to]) {
int d = dfs(e.to,t,min(f,e.cap));
if (d > 0) {
e.cap -= d;
G[e.to][e.rev].cap += d;
return d;
}
}
}
return 0;
}
//求解从s到t的最大流
int max_flow(int s,int t){
int flow = 0;
for (;;) {
bfs(s);
if (level[t] < 0) return flow;
mmm(iter,0);
int f;
while ((f = dfs(s,t,INF)) > 0)
flow += f;
}
}
int pic[MAXN][MAXN];
int dis[MAXN];
struct Node{
int to,cost;
Node(int to = 0,int cost = 0):to(to),cost(cost){}
bool operator < (const Node&a) const {
return a.cost < cost;
}
};
priority_queue<Node> qu;
vector<Node> E[MAXN];
void init(){
rep(i,1,N) E[i].clear();
rep(i,1,N) G[i].clear();
fill(dis+1,dis+1+N,INF);
mmm(pic,0);
}
void Djs(int s,int t) {
while (!qu.empty()) qu.pop();
qu.push(Node(s,0));
dis[s] = 0;
while (!qu.empty()) {
Node k = qu.top();qu.pop();
rep(i,0,E[k.to].size()-1) {
int to = E[k.to][i].to;
int cost = E[k.to][i].cost;
if (dis[to] > dis[k.to] + cost) {
dis[to] = dis[k.to] + cost;
qu.push(Node(to,dis[to]));
}
}
}
}
int main()
{
int T;
scanf("%d",&T);
while (T --) {
scanf("%d %d",&N,&M);
init();
rep(i,1,M) {
int u,v,c;
scanf("%d %d %d",&u,&v,&c);
if (u == v) continue;
E[u].pb(Node(v,c));
}
int s,t;scanf("%d %d",&s,&t);
Djs(s,t);
rep(i,1,N) {
rep(j,0,E[i].size()-1) {
int to = E[i][j].to;
int cost = E[i][j].cost;
if (dis[to] == dis[i] + cost) pic[i][to] ++;
}
}
rep(i,1,N) {
rep(j,1,N) {
if (pic[i][j]) add_edge(i,j,pic[i][j]);
}
}
printf("%d\n",max_flow(s,t));
}
}