码迷,mamicode.com
首页 > 其他好文 > 详细

【刷题】【树形dp】

时间:2019-10-10 12:52:54      阅读:64      评论:0      收藏:0      [点我收藏+]

标签:影响   mes   次数   好的   mem   sizeof   return   define   直接   

1>树上染色

(题解复制自luogu)

题目要求将k个点染成黑色,求黑点两两距离及白点两两距离,使他们之和最大。

我们可以将距离转化为路径,然后再将路径路径拆分成边,就可以记录每条边被经过的次数,直接计算即可。

很简单对吧?那么问题来了,距离转化为路径好理解,路径拆为边也好说,可是每条边被经过的次数怎么计算呢?

我们可以这样想,我们任意取两个同色的点,对于每一条边,若不在这两个点的路径上,我们自然不考虑,若是在两个点的路径上,那么这条边的计数加一。我们可以转换一下,若是两个点在边的一侧,则不影响计数,若在边的两侧,则边的计数加一。那么我们推广一下,便可以得出,一条边的两侧每有一对同色点,这条边就要被经过一次。也就是说,一条边被经过的次数等于边的两侧同色点个数的乘积。那么我们便可以求出每条边被经过的次数

tot=k*(m-k)+(sz[v]-k)*(n-m-sz[v]+k)tot=k(mk)+(sz[v]k)(nmsz[v]+k)

mm表示题目要求选的黑点数,sz[v]sz[v]表示当前子节点的子树大小,kk表示当前子节点的子树上已选择的黑点数

有了这个结论,我们就可以轻松地得出DP方程了。

f[u][j]=max(f[u][j],f[u][j-k]+f[v][k]+tot*e[i].w)f[u][j]=max(f[u][j],f[u][jk]+f[v][k]+tote[i].w)

就是关于这个方程让我在做题的时候纠结了好久,为什么kk正序排列就是对的,倒序排列就是错的?已有的题解也没有做出很好的解释,我A了之后也没有继续研究。多亏了帮同学找树形DP入门题时我重新注意到了这道题,使我对这一奇怪的现象产生了疑惑。得到了DDOSvoid大佬的帮助并进行了多次试验后,我终于明白了其中的原因,也让我对这道题的理解加深了数层。

这道题kk前几篇题解必须正序枚举的原因并不是什么要用j-kjk更新答案,而是因为正序枚举kk是从00开始的,而这道题的状态转移必须要先将k=0k=0的状态转移过来才能成立。也就是说,这只是个巧合,jj的枚举要倒序没错,但kk的枚举必须正序简直就是无稽之谈。要想避免这一情况,只需提前转移一下k=0k=0的情况即可。

#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<vector>
#include<cstring>
#define ll long long
using namespace std;
ll n,k;
const int N=2003;
struct node
{
    int v,w;
    node(int vv,ll ww)
    { v=vv,w=ww; }
    node(){}
};
vector <node> g[N];

ll sz[N];
ll f[N][N];
void dfs(int rt,int fa)
{
    sz[rt]=1,f[rt][0]=f[rt][1]=0;
    //不合法的状态的值可能很大,所以还是memset 
    int s=g[rt].size();
    for(int i=0;i<s;i++)
        if(g[rt][i].v !=fa) 
            dfs(g[rt][i].v ,rt),sz[rt]+=sz[g[rt][i].v ]; 
    
    for(ll i=0;i<s;i++)
    {
        node nx=g[rt][i];
        if(nx.v ==fa) continue;
        
    for(ll j=min(sz[rt],k);j>=0;j--)//下面只是因为好写才倒序 //不是,不这么写,就会状态重复计算
            for(ll l=0;l<=sz[nx.v ] && l<=j;l++)
            {
                if(f[rt][j-l]==-1) continue;
                
                ll val=nx.w *l*(k-l) +nx.w *(sz[nx.v ]-l)*(n-k-sz[nx.v ]+l);
                f[rt][j]=max(f[rt][j],f[rt][j-l]+f[nx.v ][l]+val);
            }
    }
}

int main()
{
    scanf("%d%d",&n,&k);
    k=min(k,n-k);
    int u,v;ll w;
    for(int i=1;i<n;i++)
    {
        scanf("%d%d%lld",&u,&v,&w);
        g[u].push_back(node(v,w)) ;
        g[v].push_back(node(u,w));
    }
    
    memset(f,-1,sizeof(f));
    dfs(1,-1);
    printf("%lld\n",f[1][k]);
    
    return 0; 
}

2>数字转换

 

 

【刷题】【树形dp】

标签:影响   mes   次数   好的   mem   sizeof   return   define   直接   

原文地址:https://www.cnblogs.com/xwww666666/p/11647104.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!