#155-[最长路]挖地雷

Description

在一个地图上有N个地窖（N<=200）,每个地窖中埋有一定数量的地雷。同时，给出地窖之间的连接路径，并规定路径都是单向的,也不存在可以从一个地窖出发经过若干地窖后又回到原来地窖的路径。某人可以从任一处开始挖地雷，然后沿着指出的连接往下挖（仅能选择一条路径），当无连接时挖地雷工作结束。设计一个挖地雷的方案，使他能挖到最多的地雷。

Input

 N                                               //地窖的个数
 W1，W2，……WN                    //每个地窖中的地雷数
 X1，Y1                                     //表示从X1可到Y1
 X2，Y2
 ……
 0，0                           

Output

K1-K2-…-Kv                             //挖地雷的顺序
MAX                                        //最多挖出的地雷数

Sample Input

6
5 10 20 5 4 5
1 2
1 4
2 4
3 4
4 5 
4 6
5 6
0 0

Sample Output

3-4-5-6
34

#include <iostream>
#include <queue>

#define SIZE 210

using namespace std;

struct edge // 表示一条边的数据结构
{
	int to, dis;
};

queue<int> q;
vector<edge> graph[SIZE];
int a[SIZE], dis[SIZE], pre[SIZE];
bool flag[SIZE], inqueue[SIZE];

void spfa(void) // 最长路SPFA
{
	int u, v, d, i;
	
	q.push(0);
	while (!q.empty())
	{
		u = q.front();
		q.pop();
		inqueue[u] = false;
		for (i = 0; i < graph[u].size(); ++i)
		{
			v = graph[u][i].to;
			d = graph[u][i].dis;
			if (dis[u] + d > dis[v])
			{
				dis[v] = dis[u] + d;
				pre[v] = u; // 在松弛时顺便设置前驱
				if (!inqueue[v])
				{
					inqueue[v] = true;
					q.push(v);
				}
			}
		}
	}
	
	return;
}

void print(int x) // 输出路径
{
	if (pre[x])
	{
		print(pre[x]);
	}
	printf("%s%d", (pre[x]) ? "-" : "", x);
	
	return;
}

int main(void)
{
	int n, i, x, y, _max = 0, index;
	
	scanf("%d", &n);
	for (i = 1; i <= n; ++i)
	{
		scanf("%d", &a[i]);
	}
	while (scanf("%d%d", &x, &y), ((x) || (y)))
	{
		graph[x].push_back({y, a[y]}); // 建边
		flag[y] = true;
	}
	
	for (i = 1; i <= n; ++i)
	{
		if (!flag[i])
		{
			graph[0].push_back({i, a[i]}); // 超级源点连接每个没有入度的点
		}
	}
	spfa();
	for (i = 1; i <= n; ++i)
	{
		if (_max < dis[i]) // 找到结果最大的一个点
		{
			_max = dis[i];
			index = i;
		}
	}
	
	print(index);
	printf("\n%d", _max);
	
	return 0;
}