scau 19082 中位特征值

Description

【2022】贝壳找房秋招测试开发工程师笔试卷2

给你一棵以T为根，有n个节点的树。（n为奇数）每个点有一个价值V，并且每个点有一个特征值P。
每个点的特征值P为：以这个点为根的子树的所有点（包括根）的价值的和。
现在牛牛想知道这n个点对应的特征值的中位数是多少，你能告诉牛牛吗？

输入格式

第一行两个正整数，分别代表T和n。n<=1e5
接下来一行共n个正整数，分别代表编号为i的点的价值V[i]。V[i]<=1e9
接下来n-1行，每行两个正整数u,v，代表u和v之间有一条边相连。

输出格式

输出一行，共一个正整数，代表n个点特征值的中位数是多少。

输入样例

2 5
1 10 100 1000 10000
1 2
3 2
3 4
5 3

输出样例

10000

提示

根据题意，2是树根，可以先画出这棵树理解题意。点1对应的特征值为1，点2对应的特征值为11111，
点3对应的特征值为11100，点4对应的特征值为1000，点5对应的特征值为10000，中位数是10000。
提示，数据范围为1e9，那么求和计算很可能超过int范围，因此数据定义时要使用long long 类型。

 根据输入样例，我们可以得到这样一棵树，1的特征值为1,4的特征值为1000,5的特征值为10000（叶子节点的特征值就是他自己），3的特征值为11100（100+1000+10000），2的特征值为11111（1+11100+10）

根据输入格式的最后一行“接下来n-1行，每行两个正整数u,v，代表u和v之间有一条边相连。”，我们可以用邻接表来创建无向图。教材上用的是链表，老师教了我们一种更为简单的方法——vector法。

vector是C++提供的动态数组，用其存储图结构占用空间为图中边数量。
语句 vector<int>a  和int a[10]作用类似，但vector定义的数组a，其大小可变，即可以自由拓展存储空间。

vector <int> e[100005];
for(int i=1;i<n;i++)
    {
        cin >> x >> y;
        e[x].push_back(y);
        e[y].push_back(x);
    }

 vector的数据不能直接写，要用push_back函数，写法如上。

每个点的价值用数组a[]来记录，这里用的是邻接表，为了避免出现特征值重复调用计算的问题我们再引用一个标记数组v[]。特征值用数组ans[]存储。

long long dfs(int cur)
{
    v[cur]=1;
    ans[cur]=a[cur];
    for(int i=0;i<e[cur].size();i++)
    {
        int y=e[cur][i];
        if(v[y]==0)//cur …… i有边
            ans[cur]+=dfs(y);
    }
    return ans[cur];
}

上图为邻接表，但是我们知道，叶子节点的特征值是他自己，不需要加上邻接点的价值，所以我们需要一个标记数组来标记，如果一个点的邻接点已经被标记了，说明该点是这个邻接点的子结点，就不需要调用函数求特征值。最后返回该点的特征值即可。这个调用应该还是很好理解的叭，我也不知道还要讲啥了哈哈哈，要是有啥疑问尽管在评论区里问就好啦！

下面是全部代码。

#include <iostream>
#include <cstdlib>
#include <cstdio>
#include <string>
#include <algorithm>
#include <vector>

using namespace std;

int T,n;
int a[100005],v[100005];
long long ans[100005];
vector <int> e[100005];

long long dfs(int cur)
{
    v[cur]=1;
    ans[cur]=a[cur];
    for(int i=0;i<e[cur].size();i++)
    {
        int y=e[cur][i];
        if(v[y]==0)//cur …… i有边
            ans[cur]+=dfs(y);
    }
    return ans[cur];
}

int main()
{
    std::ios::sync_with_stdio(false);
    cin >> T >> n;
    for(int i=1;i<=n;i++)
        cin >> a[i];
    int x,y;
    for(int i=1;i<n;i++)
    {
        cin >> x >> y;
        e[x].push_back(y);
        e[y].push_back(x);
    }
    dfs(T);
    sort(ans+1,ans+n+1);
    //for(int i=1;i<=n;i++)
        //cout << ans[i] <<' ';
        cout << ans[n/2+1];
    return 0;
}

哦对，最后要输出中位数，调用sort函数对特征值数组ans[]进行排序后再输出中位数会更方便。

在我的上一篇博客​​​里有简单介绍sort函数，有兴趣可以去看看，也可以自己在网上搜搜~