• 【BZOJ4033】[HAOI2015] 树上染色(树形DP)


    点此看题面

    大致题意: 给你一棵点数为N的带权树,要你在这棵树中选择K个点染成黑色,并将其他的N-K个点染成白色。要求你求出黑点两两之间的距离加上白点两两之间距离的和的最大值。

    树形(DP)

    这道题应该是一道比较显然的树形(DP),我们可以用f[x][i]来表示当前节点为x时有i个黑色节点时能取得的最大值。则转移方程应为(伪代码)

    f[x][i]=max(f[x][i],f[x][i-j]+f[x的一个子节点][j]+j*(m-j)*x与这个子节点之间边的边权+1LL*(Size[x的子节点]-j)*(n-m+j-Size[x的子节点])*边的边权);
    

    既然推出了转移方程,那么这题就好做了,只要先用dfs预处理,再DP就可以了。

    代码

    #include<bits/stdc++.h>
    #define max(a,b) ((a)>(b)?(a):(b))
    #define min(a,b) ((a)<(b)?(a):(b))
    #define LL long long
    #define N 2000 
    using namespace std; 
    int n,m,ee=0,lnk[N+5],fa[N+5],vis[N+5],Size[N+5];
    LL f[N+5][N+5];
    struct edge
    {
    	int to,nxt,val;
    }e[2*N+5];
    inline char tc()
    {
    	static char ff[100000],*A=ff,*B=ff;
    	return A==B&&(B=(A=ff)+fread(ff,1,100000,stdin),A==B)?EOF:*A++; 
    }
    inline void read(int &x)
    {
    	x=0;int f=1;char ch;
    	while(!isdigit(ch=tc())) if(ch=='-') f=-1;
    	while(x=(x<<3)+(x<<1)+ch-'0',isdigit(ch=tc()));
    	x*=f;
    }
    inline void write(LL x)
    {
    	if(x<0) putchar('-'),x=-x;
    	if(x>9) write(x/10);
    	putchar(x%10+'0');
    }
    inline void add(int x,int y,int z)
    {
    	e[++ee]=(edge){y,lnk[x],z},lnk[x]=ee;
    }
    inline int dfs(int x)//dfs预处理出每个节点的父亲以及子树大小
    {
    	register int i;
    	for(Size[x]=1,i=lnk[x];i;i=e[i].nxt)
    		if(e[i].to^fa[x]) fa[e[i].to]=x,Size[x]+=dfs(e[i].to);
    	return Size[x];
    }
    inline void dp(int x)//DP的核心内容
    {
    	vis[x]=1;//标记已访问
    	register int y,i,j;
    	for(y=lnk[x];y;y=e[y].nxt)
    	{
    		if(vis[e[y].to]) continue;
    		dp(e[y].to);
    		for(i=min(Size[x],m);i>=0;--i)
    			for(j=0;j<=min(Size[e[y].to],i);++j)
    				f[x][i]=max(f[x][i],f[x][i-j]+f[e[y].to][j]+1LL*j*(m-j)*e[y].val+1LL*(Size[e[y].to]-j)*(n-m+j-Size[e[y].to])*e[y].val);//转移方程
    	}
    }
    int main()
    {
    	freopen("a.in","r",stdin);
    	register int i;int x,y,z;
    	for(read(n),read(m),i=1;i<n;++i) 
    		read(x),read(y),read(z),add(x,y,z),add(y,x,z);
    	dfs(1);
    	2*m>n?m=n-m:0;//一个非常重要的优化,没有可能会TLE,原理在于在树上将m个节点染成黑色与将n-m个节点染成黑色其实是等价的
    	memset(f,167,sizeof(f));
    	for(i=1;i<=n;++i) 
    		f[i][0]=f[i][1]=0;
    	return dp(1),write(f[1][m]),0;
    }
    
  • 相关阅读:
    解决VS2012新建MVC4等项目时,收到加载程序集“NuGet.VisualStudio.Interop…”的错误
    BOM 清除
    sleep和Sleep区别
    出现error: stray ‘357’ in program的根源
    vi 复制或剪切多行超级强大方法
    CentOS乱码解决方法
    VBS实现文本文件按行数拆分的脚本
    python-门户应用状态检查脚本
    HTA+VBS实现的话单核查小工具
    oracle 11g 安装
  • 原文地址:https://www.cnblogs.com/chenxiaoran666/p/BZOJ4033.html
Copyright © 2020-2023  润新知