• CodeForces 767C Garland


    树形$dp$。

    先看权值之和是否为$3$的倍数,如果不是则一定无解。

    如果是$3$的倍数,可以分两次去切。每次一个节点,要求这个节点不是根,并且的子树权值和为$sum/3$,又要是某链上深度最深的。

    找不到两个依然是无解,否则就有解。

    #pragma comment(linker, "/STACK:1024000000,1024000000")
    #include<cstdio>
    #include<cstring>
    #include<cmath>
    #include<algorithm>
    #include<vector>
    #include<map>
    #include<set>
    #include<queue>
    #include<stack>
    #include<ctime>
    #include<iostream>
    using namespace std;
    typedef long long LL;
    const double pi=acos(-1.0);
    void File()
    {
        freopen("D:\in.txt","r",stdin);
        freopen("D:\out.txt","w",stdout);
    }
    template <class T>
    inline void read(T &x)
    {
        char c = getchar();
        x = 0;
        while(!isdigit(c)) c = getchar();
        while(isdigit(c))
        {
            x = x * 10 + c - '0';
            c = getchar();
        }
    }
    
    int n,root,c[1000010],p[1000010],sum,ans1,ans2;
    vector<int >v[1000010];
    int flag1,flag2;
    
    void dfs1(int x)
    {
        c[x]=p[x];
        for(int i=0;i<v[x].size();i++)
        {
            dfs1(v[x][i]); if(flag1) return ;
            c[x]+=c[v[x][i]];
        }
        if(x!=root&&c[x]==sum/3)
        {
            flag1=1;
            ans1=x;
        }
    }
    
    void dfs2(int x)
    {
        c[x]=p[x];
        for(int i=0;i<v[x].size();i++)
        {
            if(v[x][i]==ans1) continue;
            dfs2(v[x][i]); if(flag2) return ;
            c[x]+=c[v[x][i]];
        }
        if(x!=root&&c[x]==sum/3)
        {
            flag2=1;
            ans2=x;
        }
    }
    
    int main()
    {
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
        {
            int fa; scanf("%d%d",&fa,&p[i]);
            sum=sum+p[i];
            if(fa==0) root=i;
            else v[fa].push_back(i);
        }
    
        if(sum%3!=0)
        {
            printf("-1
    ");
            return 0;
        }
    
        dfs1(root);
    
        if(flag1==0)
        {
            printf("-1
    ");
            return 0;
        }
    
        memset(c,0,sizeof c);
        dfs2(root);
    
        if(flag2==0)
        {
            printf("-1
    ");
            return 0;
        }
    
        else
        {
            printf("%d %d
    ",ans1,ans2);
        }
    
        return 0;
    }
  • 相关阅读:
    CF1592F2 Alice and Recoloring 2
    CF1601E Phys Ed Online
    AGC050B Three Coins
    [学习笔记]珂朵莉树(Old Drive Tree)
    CF30E. Tricky and Clever Password
    [学习笔记]替罪羊树
    开源项目MiniOA队员招募通知
    MiniOA开发过程记录(33)自动登录模式
    MiniOA开发过程记录(29)安装Maven
    简易工作流设计思考(欢迎补充和批评)
  • 原文地址:https://www.cnblogs.com/zufezzt/p/6415495.html
Copyright © 2020-2023  润新知