cf 1401D - Maximum Distributed Tree

最新推荐文章于 2023-07-26 19:44:22 发布

Eriii34

最新推荐文章于 2023-07-26 19:44:22 发布

阅读量92

点赞数

分类专栏： cf题文章标签： c语言算法

本文链接：https://blog.csdn.net/Eriii/article/details/120223805

版权

cf题专栏收录该内容

14 篇文章 0 订阅

订阅专栏

本文探讨了如何解决CF 1401D问题，通过深搜策略记录节点数量，并结合贪心算法，将最大乘积分配给影响答案最多的边。

摘要由CSDN通过智能技术生成

cf 1401D - Maximum Distributed Tree

题目链接
做法：深搜（记录当前边下节点个数，来统计对答案产生贡献的次数） + 贪心（将最大的乘积赋值给对于答案产生最多次贡献的边，其他顺次给）

#include<bits/stdc++.h>
using namespace std;

typedef long long ll;
typedef pair<int, int> pii;
typedef pair<ll, int> pli;

const int N= 1e5+10;
const int mod = 1e9+7;

map<pii, int> mp;
vector<int> son[N];
pli w[N];
int p[N];
int n, m;

void dfs(int u, int fa) {
    int e = mp[{max(u, fa), min(u, fa)}];
    w[e].first = 1;
    for(auto v : son[u]) {
        if(v == fa) continue;
        dfs(v, u);
        int ee = mp[{max(u, v), min(u, v)}];
        w[e].first += w[ee].first;
    }
}

int Plus(ll a, ll b) {
    return a + b >= mod ? a+b - mod : a+b;
}

int Mul(ll a, ll b) {
    return a * b % mod;
}

void init() {
    for(int i = 1; i <= n + 1; i++) son[i].clear();
    mp.clear();
}

int main() {
    int T;
    scanf("%d", &T);
    while(T -- ) {
        scanf("%d", &n);
        n--;
        for(int i = 1; i <= n; i++) {
            int u, v;
            scanf("%d%d", &u, &v);
            mp[{max(u, v), min(u,v)}] = i;
            son[u].push_back(v);
            son[v].push_back(u);
            w[i].second = i;
        }
        scanf("%d", &m);
        for(int i = 1; i <= m; i++) scanf("%d", &p[i]);
        sort(p+1, p+1+m);

        dfs(1, 0);
        for(int i = 1; i <= n; i++) {
            w[i].first = w[i].first * (n+1 - w[i].first);
        }
        sort(w+1, w+1+n);
        ll ans = 0;
        if(m < n) {
            int i = n, j = m;
            for( ; j; i--, j --) {
                ans =Plus(ans, Mul(w[i].first, p[j]));
            }
            while(i) {
                ans = Plus(ans, Mul(w[i].first, 1));
                i--;
            }
        }
        else {
            int i ,j;
            for(i = 1, j = 1; i < n; i++, j++) {
                ans = Plus(ans, Mul(w[i].first, p[j]));
            }
            ll add = w[n].first;
            for(j; j <= m; j++) {
                add = Mul(add, p[j]);
            }
            ans = Plus(add, ans);
        }
        // printf("********");
        printf("%lld\n", ans);
        init();
    }
    return 0;
}