E. Tree Shuffling(树上问题)

题目传送门

题意: 给你一颗树,根节点是1,每个节点有一个代价a[i],我们每个点有一个原始数字b[i],还有一个目标数字c[i](b[i],c[i]∈[0,1]),你可以挑选一个节点,然后给这个节点的k个子节点(0<=k<=子树大小)的原始数字任意排序,这个操作的代价是k*a[i],现在要使所有节点的数字都转变成目标数字,最小代价是多少?

思路: 我们dfs,维护一颗子树中的1和0的个数,并且向下维护a[i]的最小值,如果这时候a[i]=mi,那么说明他的祖先节点中没有比他小的,所以他的子孙节点靠他来重新排列是花费最小的。每次操作完之后就减去被修改的点的个数,最后判断一下是-1的情况就行。

#include
#define endl '\n'
#define null NULL
#define ls p<<1
#define rs p<<1|1
#define fi first
#define se second
#define mp make_pair
#define pb push_back
#define ll long long
#define int long long
#define pii pair
#define ull unsigned long long
#define all(x) x.begin(),x.end()
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define ct cerr<<"Time elapsed:"<<1.0*clock()/CLOCKS_PER_SEC<<"s.\n";
char *fs,*ft,buf[1<<20];
#define gc() (fs==ft&&(ft=(fs=buf)+fread(buf,1,1<<20,stdin),fs==ft))?0:*fs++;
inline int read(){int x=0,f=1;char ch=gc();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=gc();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=gc();}
return x*f;}
using namespace std;
const int N=2e5+5;
const int inf=0x7fffffff;
const int mod=998244353;
const double eps=1e-20;
const double PI=acos(-1);
int a[N],b[N],c[N];
vector<int>e[N];
int cnt0[N],cnt1[N],res=0;
void dfs(int x,int y,int mi)
{
    mi=min(mi,a[x]);
    for(auto i:e[x])
    {
        if(i!=y)
        {
            dfs(i,x,mi);
            cnt0[x]+=cnt0[i];
            cnt1[x]+=cnt1[i];
        }
    }
    if(b[x]!=c[x])
    {
        if(b[x]==1)
            cnt1[x]++;
        else
            cnt0[x]++;
    }
    if(mi==a[x])
    {
        int s=min(cnt0[x],cnt1[x]);
        res+=s*2*a[x];
        cnt1[x]-=s;
        cnt0[x]-=s;
    }
}
signed main()
{
        int n;
        cin>>n;
        for(int i=1;i<=n;i++)
        {
            cin>>a[i]>>b[i]>>c[i];
        }
        for(int i=1;i<=n-1;i++)
        {
            int u,v;
            cin>>u>>v;
            e[u].pb(v);
            e[v].pb(u);
        }
        dfs(1,1,inf);
        if(cnt0[1]||cnt1[1])
        {
            cout<<-1<<endl;
        }
        else
            cout<<res<<endl;
}

你可能感兴趣的:(树上问题)