美团点评笔试题-图的遍历

题目描述：
给定一张包含N个点，N-1条边的无向图，节点从1到N编号，每条边的长度均为1.假设你从1号节点出发并打算遍历所有节点，那么总路程至少是多少？
输入：
第一行包含一个整数N，$1\leq$ $N\leq10^5$ 。
接下来N-1行，每行包含两个整数X和Y，表示X号节点和Y号节点之间的一条边，$1\leq X$ 、$1\leq 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;
}