Invitation Cards

http://acm.hdu.edu.cn/showproblem.php?pid=1535

题意：有编号为1~p的站点，有Q条公交路线，公交车路线只能从起点走到对应的终点，而且是单向的，每条路线有其对应的车费。早上有p个人从站点1出发，到达与其编号对应的站点（编号为i的人到达编号为i的站点），晚上从各自站点回到1站点，问他们来回的最小费用之和。

 

思路：很水的题。

一个是求源点到其他点最短路，一个是求多源点到单点最短路，第二种反向建图，转化为单源点到多点最短路。这里边数有100W,所以邻接表+spfa,用Dijstra会超时。

 

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<vector>
#include<queue>
using namespace std;

const int maxn = 1000010;
const int INF = 0x3f3f3f3f;
struct node
{
	int v,w;
};
vector <struct node> edge[maxn];
int n,m,ans;
int dis[maxn],inque[maxn];
int u[maxn],v[maxn],w[maxn];

void spfa(int s)
{
    for(int i = 1; i <= n; i++)
        dis[i] = INF;
    memset(inque,0,sizeof(inque));
    queue<int> que;
    while(!que.empty())
        que.pop();

    inque[s] = 1;
    dis[s] = 0;
    que.push(s);

    while(!que.empty())
    {
        int u = que.front();
        que.pop();
        inque[u] = 0;

        for(int i = 0; i < (int)edge[u].size(); i++)
        {
            int v = edge[u][i].v;
            int w = edge[u][i].w;
            if(dis[v] > dis[u] + w && dis[u] < INF)
            {
                dis[v] = dis[u] + w;
                if(inque[v] == 0)
                {
                    inque[v] = 1;
                    que.push(v);
                }
            }
        }
    }
}

int main()
{
	int test;
	scanf("%d",&test);
	while(test--)
	{
		ans = 0;
		scanf("%d %d",&n,&m);
		for(int i = 1; i <= n; i++)
			edge[i].clear();
		for(int i = 0; i < m; i++)
		{
			scanf("%d %d %d",&u[i],&v[i],&w[i]);
			edge[u[i]].push_back((struct node){v[i],w[i]});
		}
		spfa(1);
		for(int i = 1; i <= n; i++)
			ans += dis[i];
		for(int i = 1; i <= n; i++)
			edge[i].clear();
		for(int i = 0; i < m; i++)
			edge[v[i]].push_back((struct node){u[i],w[i]});
		spfa(1);
		for(int i = 1; i <= n; i++)
			ans += dis[i];
		printf("%d\n",ans);
	}
	return 0;
}