ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

Codeforces 348B - Apple Tree

2019-02-04 15:47:46  阅读:274  来源: 互联网

标签:lcm const int LL Tree Codeforces dv 348B define


348B - Apple Tree

我们设最后答案为 x , 我们我们就能用x表示出所有节点下面的苹果个数, 然后用叶子节点求lcm, 取最大的可行解。

#include<bits/stdc++.h>
#define LL long long
#define fi first
#define se second
#define mk make_pair
#define PLL pair<LL, LL>
#define PLI pair<LL, int>
#define PII pair<int, int>
#define SZ(x) ((int)x.size())
#define ull unsigned long long
using namespace std;

const int N = 1e5 + 7;
const int inf = 0x3f3f3f3f;
const LL INF = 0x3f3f3f3f3f3f3f3f;
const int mod = 1e9 + 7;
const double eps = 1e-8;

int n, a[N];
LL dv[N], sum[N], all;
vector<int> leaf;
vector<int> G[N];

void solve(int u, int fa) {
    if(u != 1 && SZ(G[u]) == 1) leaf.push_back(u);
    sum[u] = a[u];
    for(int v : G[u]) {
        if(v == fa) continue;
        solve(v, u);
        sum[u] += sum[v];
    }
}

void dfs(int u, int fa) {
    if(dv[u] > all) {
        printf("%lld\n", all);
        exit(0);
    }
    for(int v : G[u]) {
        if(v == fa) continue;
        if(u == 1) dv[v] = dv[u] * SZ(G[u]);
        else dv[v] = dv[u] * (SZ(G[u]) - 1);
        dfs(v, u);
    }
}

bool check(LL x) {
    for(auto& id : leaf) {
        if(x / dv[id] > a[id]) return false;
    }
    return true;
}
int main() {
    scanf("%d", &n);
    for(int i = 1; i <= n; i++) scanf("%d", &a[i]), all += a[i];
    for(int i = 1; i < n; i++) {
        int u, v; scanf("%d%d", &u, &v);
        G[u].push_back(v);
        G[v].push_back(u);
    }
    dv[1] = 1;
    solve(1, 0);
    dfs(1, 0);
    LL lcm = 1;
    for(auto& id : leaf) {
        LL gcd = __gcd(lcm, dv[id]);
        if(1.0 * lcm / gcd * dv[id] > all) {
            printf("%lld\n", all);
            return 0;
        }
        lcm = lcm / gcd * dv[id];
    }
    LL low = 1, high = all / lcm, ans = 0;
    while(low <= high) {
        LL mid = low + high >> 1;
        if(check(mid * lcm)) ans = mid, low = mid + 1;
        else high = mid - 1;
    }
    printf("%lld\n", all - ans * lcm);
    return 0;
}

/*
*/

 

标签:lcm,const,int,LL,Tree,Codeforces,dv,348B,define
来源: https://www.cnblogs.com/CJLHY/p/10351790.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有