E. Tree Shuffling——(树形DP+贪心)

总结

自己当时赛中的问题,没想太清楚就是开始码了,自己的写法,虽然也是贪心,但是两次DFS,代码大,先排序从小到大,搜未搜索的子树,然后把多余的点往上移动。
第一次DFS:确定爸爸
第二次DFS:更新子树的最优值

当看了别人的代码后,就知道自己有些多余了,每个结点的如果需要向目标值改变,需要的是祖祖辈辈最小的点权,DFS更新下去就可以了,我还傻乎乎的不断贪心求子树。

算法核心

树形DP:
1:DFS每次得知道爸爸
2:递归:祖辈(root->父节点的简单路径)的关系
3:回溯:子孙(子树)的关系

题目链接

//#pragma GCC optimize(2)
//#pragma GCC target ("sse4")
#include
//typedef long long ll;
#define ull       unsigned long long
#define int       long long
#define F           first
#define S           second
#define endl        "\n"//<
#define eps         1e-6
#define base        131
#define lowbit(x)   (x&(-x))
#define PI          acos(-1.0)
#define inf         0x3f3f3f3f
#define MAXN        0x7fffffff
#define INF         0x3f3f3f3f3f3f3f3f
#define ferma(a,b)  pow(a,b-2)
#define mod(x)      (x%mod+mod)%mod
#define pb          push_back
#define decimal(x)  cout << fixed << setprecision(x);
#define all(x)      x.begin(),x.end()
#define memset(a,b) memset(a,b,sizeof(a));
#define IOS         ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
using namespace std;
void file()
{
#ifdef ONLINE_JUDGE
#else
    freopen("D:/LSNU/codeforces/duipai/data.txt","r",stdin);
    //  freopen("D:/LSNU/codeforces/duipai/WA.txt","w",stdout);
#endif
}
const int N=2e5+5;
int dp[N][2],a[N],ans=0;
vector<int>G[N];
struct node
{
    int a,b,c;
}s[N];
void dfs(int u,int fat)
{
    s[u].a=min(s[u].a,s[fat].a);
    if(s[u].b!=s[u].c)
        dp[u][s[u].b]++;
    for(auto v:G[u])
    {
        if(v==fat)
            continue;
        dfs(v,u);
        dp[u][0]+=dp[v][0];
        dp[u][1]+=dp[v][1];
    }
    int sum=min(dp[u][0],dp[u][1]);
    dp[u][1]-=sum;
    dp[u][0]-=sum;
    ans+=s[u].a*sum*2;
}
signed main()
{
    IOS;
    file();
    int n;
    cin>>n;
    int cnt=0;
    for(int i=1;i<=n;i++)
    {
        cin>>s[i].a>>s[i].b>>s[i].c;
        if(s[i].b!=s[i].c)
            cnt+=(s[i].b==0?1:-1);
    }
    s[0].a=inf;
    for(int i=1;i<n;i++)
    {
        int u,v;
        cin>>u>>v;
        G[u].pb(v);
        G[v].pb(u);
    }
    dfs(1,0);
    cout<<(cnt?-1:ans)<<endl;






    return 0;
}

你可能感兴趣的:(#,树形DP)