题目描述:
给定一张包含N个点,N-1条边的无向图,节点从1到N编号,每条边的长度均为1.假设你从1号节点出发并打算遍历所有节点,那么总路程至少是多少?
输入:
第一行包含一个整数N, 1≤ 1 ≤ N≤105 N ≤ 10 5 。
接下来N-1行,每行包含两个整数X和Y,表示X号节点和Y号节点之间的一条边, 1≤X 1 ≤ X 、 1≤Y 1 ≤ Y 。
输出:
输出总路程的最小值。
样例输入:
4
1 2
1 3
3 4
样例输出:
4
Hint:
按1->2->1->3->4的路线遍历所有节点,总路程为4.
解题思路:
根据题意以1节点为出发点可能有多条路径,输出总路程的最小值 = 1节点出发的最长路径 + 1节点出发的其他路径 * 2
#include <iostream>
#include <vector>
#include<algorithm>
using namespace std;
#define max_num 100000
int cost(int i, int j, int n, int **arr)
{
int w_sum = 0;
while (i < n - 1 && j < n)
{
if (i!=j && arr[i][j]!=max_num)
{
w_sum += 1;
i = j;
j++;
}
else
{
j++;
}
}
return w_sum;
}
int main()
{
int N;
int link_num=0;
int *costed;
cin >> N;
int **arr = new int*[N];
for (int i = 0; i < N; i++)
{
arr[i] = new int[N];
for (int j = 0; j < N; j++)
{
if (i == j)
{
arr[i][i] = 0;
}
else
{
arr[i][j] = max_num;
}
}
}
for (int i = 0; i < N-1; i++)
{
int p, q;
cin >> p >> q;
arr[p - 1][q - 1] = arr[q - 1][p - 1] = 1;
}
for (int i = 0; i < N; i++)
{
if (arr[0][i]==1)
{
link_num += 1;
}
}
costed = new int[N];
for (int i = 0; i < N; i++)
{
if (arr[0][i] == 1)
{
costed[i] = cost(0, i, N, arr);
}
else
{
costed[i] = 0;
}
}
sort(costed,costed+N);
int short_sum=0;
for (int i = 0; i < N; i++)
{
if (i<N-1)
{
short_sum += costed[i] * 2;//非最长路径都要走两遍,这里乘于2
}
else
{
short_sum += costed[i];//最长路径只需要走一遍
}
}
cout << short_sum;
return 0;
}