sgu 185 Two shortest 最大流

这道题思路就是由最短路边组成的边集搞一次最大流得到的是互不相交的路径。

首先用SPFA得到各点到源点的最短距离，在遍历一遍所有的边，通过判断是否满足dis[u]+map[u][v]==dis[v]来判断是否在最短路边集上。

然后用SAP求一下从1到n的最大流，如果小于2则no solution。否则dfs2次打印路径。

#include<stdio.h>
#include<stdlib.h>
#include<algorithm>
#include<string.h>
#include<queue>
#define min(a,b) ((a)>(b))?(b):(a)
using namespace std; 
#define INF 1e9
int map[402][402];
int dis[402];
int n,m,flag;
int k,dist[402],gap[402],edgehead[402];//双向边
struct{
    int v, cap, next, re;
}edge[160002];

void addedge(int u,int v,int ca){
    edge[k].v=v;
    edge[k].cap=ca;
    edge[k].next=edgehead[u];
    edge[k].re=k + 1;               //这个用来记录此边的反边
    edgehead[u]=k ++;
    edge[k].v=u;
    edge[k].cap=0;
    edge[k].next=edgehead[v];
    edge[k].re=k - 1;
    edgehead[v]=k ++;
}

int dfs (int p , int limit)
{
    if(p==n)return limit;

    for(int i=edgehead[p];i!=0;i=edge[i].next){
        int v = edge[i].v;
        if(dist[p]==(dist[v]+1) && edge[i].cap>0){
            int t=dfs(v,min(limit , edge[i].cap));
            if(t<0)return t;//没有增广路了
            if(t>0)//更新流
            {
                edge[i].cap-=t;
                edge[edge[i].re].cap+=t;
                return t;
            }
        }
    }

    int tmp=n+1 ;
    for(int i=edgehead[p];i!=0;i=edge[i].next){
        int v = edge[i].v;
        if(edge[i].cap>0)
            tmp=min(tmp,dist[v]+1);
    }

    if(--gap[dist[p]]==0 || dist[0]>n)return -1;//出现断层或回流已满
    ++gap[dist[p]=tmp];
    return 0;
}

int SAP()
{
    gap[0]=n+1;
    int f = 0 , t=0;
    while (~(t=dfs(0,INF))) f+=t;
    return f;
}

void SPFA(){
	int use[402],i;
	memset(use,0,sizeof(use));
	for(i=0;i<=n;i++)
		dis[i]=INF;
	queue<int>que;
	dis[0]=0;
	que.push(0);
	use[0]=1;
	while(!que.empty()){
		int x=que.front();
		que.pop();
		use[x]=0;
		for(i=0;i<=n;i++){
			if(map[x][i]!= INF && dis[x]+map[x][i]<dis[i]){
				dis[i]=dis[x]+map[x][i];
				if(!use[i]){
					use[i]=1;
					que.push(i);
				}
			}
		}
	}
}
void make_graph(){
	int i,j;
	k=1;
	memset(dist,0,sizeof(dist));
	memset(gap,0,sizeof(gap));
	memset(edgehead,0,sizeof(edgehead));
	for(i=0;i<=n;i++)
		for(j=0;j<=n;j++){
			if(i==j || map[i][j]==INF)
				continue;
			if(dis[i]+map[i][j]==dis[j]){
				addedge(i,j,1);
			}
		}
}
void dfs(int p){
	if(flag)
		return;
	int v;
	if(p==n){
		printf("\n");
		flag=1;
		return ;
	}
	for(int i=edgehead[p];i!=0 && !flag;i=edge[i].next){
		v=edge[i].v;
		if(edge[i].cap==0 && i%2==1){
			edge[i].cap=-1;
			printf(" %d",v+1);
			dfs(v);   //直接搜即可，因为正向边cap为0则一定有流量从s到t经过此边
		}
	}
}
int main(){
	int i,j,u,v,c;

	scanf("%d %d",&n,&m);
	n--;
	for(i=0;i<=n;i++)
		for(j=0;j<=n;j++)
			map[i][j]=INF;
	for(i=1;i<=m;i++){
		scanf("%d %d %d",&u,&v,&c);
		u--;v--;
		map[u][v]=map[v][u]=c;
	}

	SPFA();
	make_graph();

	if(SAP()>=2){
		printf("1");
		flag=0;
		dfs(0);

		printf("1");
		flag=0;
		dfs(0);
	}
	else
		printf("No solution\n");

	return 0;
}