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

题目描述:
给定一张包含N个点,N-1条边的无向图,节点从1到N编号,每条边的长度均为1.假设你从1号节点出发并打算遍历所有节点,那么总路程至少是多少?
输入:
第一行包含一个整数N, 1 1 ≤ N105 N ≤ 10 5
接下来N-1行,每行包含两个整数X和Y,表示X号节点和Y号节点之间的一条边, 1X 1 ≤ X 1Y 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;
}
    原文作者:数据结构之图
    原文地址: https://blog.csdn.net/qq_36427732/article/details/82585095
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞