描述
题解
树归问题,名字虽然叫TSP,却和TSP没有啥关系,一道典型的树归问题。当然,方法不止一种,还可以先用拓扑排序预处理一下。
代码
#include <cstdio>
#define MAX_N 100005
#define MAX_M 1000005
#define max(a, b) ((a > b) ? (a) : (b))
using namespace std;
const int INF = 0x3f3f3f3f;
struct edge
{
int to, w, next;
edge()
{
w = to = next = 0;
}
};
edge D[MAX_M]; // 邻接表
int head[MAX_N];// 邻接表头
int dp[MAX_M]; // dp[i]:第i条边为开始的最长路
int num[MAX_N]; // num[i]:以i为起点的最长路
int cnt = 1;
// 往邻接表中添加边
void add(int u, int v, int w)
{
D[cnt].to = v;
D[cnt].w = w;
D[cnt].next = head[u];
head[u] = cnt++;
}
// 树归部分
int DFS(int s)
{
int maxn = 0;
for (int i = head[s]; i; i = D[i].next)
{
if (dp[i] == -1)
{
dp[i] = DFS(D[i].to) + D[i].w;
}
maxn = max(maxn, dp[i]);
}
return maxn;
}
// 从起点开始查找最长路
void find(int s, int w)
{
printf("%d\t", s);
int minx = INF, W = 0;
for (int i = head[s]; i; i = D[i].next)
{
// 保证在最长路上并且字典序最小
if (dp[i] == w && minx > D[i].to)
{
W = w - D[i].w;
minx = D[i].to;
}
}
if (minx != INF)
{
find(minx, W);
}
return ;
}
int main()
{
int n, m;
scanf("%d%d", &n, &m);
for (int i = 0; i <= m; i++)
{
dp[i] = -1;
}
int a, b, p;
for (int i = 0; i < m; i++)
{
scanf("%d%d%d", &a, &b, &p);
add(a, b, p);
}
int maxn = -1, temp = 0;
for (int i = 1; i <= n; i++)
{
num[i] = DFS(i);
if (num[i] > maxn)
{
maxn = num[i];
temp = i;
}
}
find(temp, maxn);
printf("\n");
return 0;
}