码迷,mamicode.com
首页 > 其他好文 > 详细

HDU4607 Park Visit(解法二)

时间:2016-08-21 19:59:10      阅读:229      评论:0      收藏:0      [点我收藏+]

标签:

问题链接HDU4607 Park Visit

题意简述莱克尔和她的朋友到公园玩,公园很大也很漂亮。公园包含n个景点通过n-1条边相连。克莱尔太累了,所以不能去参观所有点景点。经过深思熟虑,她决定只访问其中的k个景点。她拿出地图发现所有景点的入口都很特殊。所以她想选择一个入口,并找到一条最短的路来参观k个景点。假设景点之间的距离为1。

输入数据:先输入测试用例数t,每个测试用例数据包括N和M,N为节点数,M为需要查询的上述的k(参观M个景点)

输出数据:对于各个M,输出需要走的距离。

问题分析:n个节点n-1条边构成无向无环树。该问题就是求树直径,树直径即树上最长路径。假设树直径包含m个节点(直径为m-1),若k<=m,则答案为k-1,否则答案为m + (k-m-1)*2。

程序说明与同题的另外一种解法不同,这个程序先随意找出一个节点(程序中是节点1),用函数dfs(),求出1节点到其他各个节点的距离,从中选出距离1节点最远的节点(程序中放在变量start中)。然后,再以节点start作为起点,函数dfs(),求出节点start到其他各个节点的距离,从中选出距离start节点最远的节点(程序中放在变量target中),那么从start到target的距离就是树直径。

相比较而言,这个程序速度上会稍微快一些。参见:HDU4607 Park Visit

AC的C++语言程序如下:

/* HDU4607 Park Visit */

#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>

using namespace std;

const int MAXN = 100000;
vector<int>tree[MAXN+1];
int maxlen, dist[MAXN+1];

void dfs(int now, int last, int d[])
{
    int u, size;
    size = tree[now].size();
    for(int i=0; i<size; i++)
        if ((u = tree[now][i]) != last) {
            d[u] = d[now] + 1;
            dfs(u, now, d);
        }
}

int main(void)
{
    int t, n, m, u, v, k;

    scanf("%d", &t);
    while(t--) {
        scanf("%d%d", &n, &m);

        for(int i=1; i<=n; i++)
            tree[i].clear();

        for(int i=1; i<n; i++) {
            scanf("%d%d", &u, &v);
            tree[u].push_back(v);
            tree[v].push_back(u);
        }

        dist[1] = 0;
        dfs(1, 0, dist);

        int start = 0;
        dist[start] = 0;
        for(int i=1; i<=n; i++)
            if(dist[i] > dist[start])
                start = i;

        dist[start] = 0;
        dfs(start, 0, dist);
        int target = 0;
        for (int i=1; i<=n; i++)
            if(dist[i] > dist[target])
                target = i;
        maxlen = dist[target];

        while(m--) {
            scanf("%d", &k);
            if(k <= maxlen)
                printf("%d\n", k-1);
            else
                printf("%d\n", maxlen + (k-maxlen-1)*2);
        }
    }

    return 0;
}


HDU4607 Park Visit(解法二)

标签:

原文地址:http://blog.csdn.net/tigerisland45/article/details/52268881

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!