poj1511

//  Created by JamesQi in 2015.
//  Copyright (c) 2015 JamesQi. All rights reserved.
//poj1511,两次spfa求和,主要是考链式前向星,用邻接表会超时,还有就是建立反向边;

#include <cstdio>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <climits>
#include <string>
#include <vector>
#include <cmath>
#include <stack>
#include <queue>
#include <set>
#include <map>

using namespace std;

const int inf=0x3f3f3f3f;
const int maxn=1e6+10;

struct Edge
{
	int v,cost,next;
}edge[maxn],sedge[maxn];

int head[maxn],hand[maxn];
int cnt;
int ans;
int dp[maxn];
int n,m;
void addedge(int u,int v,int c)
{
	edge[cnt].v=v;
	edge[cnt].cost=c;
	edge[cnt].next=head[u];
	head[u]=cnt++;

	sedge[ans].v=u;
	sedge[ans].cost=c;
	sedge[ans].next=hand[v];
	hand[v]=ans++;
	return ;
}

void spfa1()
{
	//int dp[maxn];
	bool vis[maxn];
	memset(vis,false,sizeof vis);
	memset(dp,inf,sizeof dp);
	queue<int> que;
	while(!que.empty()) que.pop();
	vis[1]=true;
	dp[1]=0;
	que.push(1);
	while(!que.empty())
	{
		int u=que.front();
		que.pop();
		vis[u]=false;
		for (int i=head[u];i!=-1;i=edge[i].next)
		{
			Edge& e=edge[i];
			if (dp[e.v]>dp[u]+e.cost)
			{
				dp[e.v]=dp[u]+e.cost;
				if (!vis[e.v])
				{
					vis[e.v]=true;
					que.push(e.v);
				}
			}
		}
	}
	/*
	int tot=0;
	for (int i=1;i<=n;i++)
		tot+=dp[i];
	//printf("spfa1=%d\n",tot);
	return tot;
	*/
	return ;
}

void spfa2()
{
	queue<int> que;
	while(!que.empty()) que.pop();
	//int dp[maxn];
	bool vis[maxn];
	memset(dp,inf,sizeof dp);
	memset(vis,false,sizeof vis);
	que.push(1);
	vis[1]=true;
	dp[1]=0;
	while(!que.empty())
	{
		int u=que.front();
		que.pop();
		vis[u]=false;
		for (int i=hand[u];i!=-1;i=sedge[i].next)
		{
			Edge& e=sedge[i];
			if (dp[e.v]>dp[u]+e.cost)
			{
				dp[e.v]=dp[u]+e.cost;
				if (!vis[e.v])
				{
					que.push(e.v);
					vis[e.v]=true;
				}
			}
		}
	}
	/*
	int tot=0;
	for (int i=1;i<=n;i++)
		tot+=dp[i];
	//printf("spfa2=%d\n",tot);
	return tot;
	*/
	return ;
}

int main()
{
	//ios::sync_with_stdio(false);

	//freopen("in.txt","r",stdin);
	//freopen("out.txt","w",stdout);
	int t;
	scanf("%d",&t);
	while(t--)
	{
		memset(head,-1,sizeof head);
		memset(hand,-1,sizeof hand);
		scanf("%d%d",&n,&m);
		cnt=ans=0;
		int a,b,c;
		for (int i=1;i<=m;i++)
		{
			scanf("%d%d%d",&a,&b,&c);
			addedge(a,b,c);
		}
		long long sum=0;
		spfa1();
		for (int i=2;i<=n;i++)
			sum+=dp[i];
		spfa2();
		for (int i=2;i<=n;i++)
			sum+=dp[i];
		printf("%lld\n",sum);
	}
	return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值