求助WA on test2
查看原帖
求助WA on test2
389955
WillW_Chen楼主2023/8/9 15:28

f[i][0/1]表示以i为根的子树在父节点不给/给多一条路径的情况的总价值

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<iomanip>
#include<algorithm>
#include<cmath>
#include<vector>
#include<bitset>
#include<list>
#include<set>
#include<queue>
#include<map>
#include<stack>
#include<ctime>
#include<random>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
#define endl "\n"
#define int long long
const ll MAXN=1e6+10;
const ll inf=0x3f3f3f3f;
const ll infll=0x3f3f3f3f3f3f3f3f;
int n,k;
ll s[MAXN];
vector<int>g[MAXN];
ll f[MAXN][2];//以u为根的子树在父亲节点给不给多一条路径的情况下的最大价值
ll num[MAXN];
bool cmp(const ll x,const ll y){
    return f[x][1]>f[y][1];
}
void init(int lim){
    for(int i=1;i<=lim;i++){
        g[i].clear();
        f[i][0]=0;
        f[i][1]=0;
        num[i]=0;
    }
    return;
}
void dfs(int u,ll tmp){
    if(!g[u].size()){
        return;
    }
    ll tmp2=tmp/g[u].size();
    for(int i=0;i<g[u].size();i++){
        int v=g[u][i];
        num[v]=tmp2;
        dfs(v,tmp2);
    }
    return;
}
void dfs2(int u){
    f[u][0]=s[u]*num[u];
    f[u][1]=s[u]*(num[u]+1);
    if(!g[u].size()){
        return;
    }
    ll lef=num[u]%g[u].size();
    vector<int>son;
    for(int i=0;i<g[u].size();i++){
        int v=g[u][i];
        dfs2(v);
        son.push_back(v);
    }
    sort(son.begin(),son.end(),cmp);
    int cnt=0;
    for(int i=0;i<son.size();i++){
        int v=son[i];
        cnt++;
        if(cnt<=lef){
            f[u][0]+=f[v][1];
        }
        else{
            f[u][0]+=f[v][0];
        }
    }
    cnt=0;
    for(int i=0;i<son.size();i++){
        int v=son[i];
        cnt++;
        if(cnt<=lef+1){
            f[u][1]+=f[v][1];
        }
        else{
            f[u][1]+=f[v][0];
        }
    }
    return;
}
void solve(){
    cin>>n>>k;
    init(n);
    for(int i=2;i<=n;i++){
        int p;
        cin>>p;
        g[p].push_back(i);
    }
    for(int i=1;i<=n;i++){
        cin>>s[i];
    }
    dfs(1,k);
    num[1]=k;
    dfs2(1);
    cout<<f[1][0]<<endl;
    return;
}
signed main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    int T;
    cin>>T;
    while(T--) solve();
    return 0;
}
2023/8/9 15:28
加载中...