洛谷题单【动态规划4】树与图上的动态规划

岁余十二.

已于 2022-09-18 11:38:28 修改

阅读量267

点赞数 1

分类专栏：洛谷文章标签：动态规划

于 2022-09-17 21:55:28 首次发布

本文链接：https://blog.csdn.net/Tracy_yi/article/details/126909544

版权

洛谷专栏收录该内容

20 篇文章 1 订阅

订阅专栏

P1122 最大子树和

思路

$d p [i]$ 表示以 $i$ 为根节点的最大子树和
对所有 $i$ 的子结点 $k$ ，如果 $d p [k] > 0$ 则 $d p [i] + = d p [k]$ ，如果 $d p [k] <= 0$ 则舍弃掉这一枝

实现

#include <bits/stdc++.h>
#define maxn 16010
using namespace std;
typedef long long ll;
int ans = -2147483647,n;
int dp[maxn]; 
vector<int> G[maxn];
void dfs(int child,int fa)
{
	int sum = G[child].size();
	for(int i=0;i<sum;i++)
	{
		if(G[child][i]!=fa)
		{
			dfs(G[child][i],child);
			if(dp[G[child][i]]>0) dp[child]+=dp[G[child][i]];	
		}
	}
}
int main()
{
	int t1,t2;
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&dp[i]);
	}
	for(int i=1;i<n;i++)
	{
		scanf("%d %d",&t1,&t2);
		G[t1].push_back(t2);
		G[t2].push_back(t1);
	}
//	for(int i=1;i<=n;i++)
//	{
//		for(int j=0;j<G[i].size();j++)
//		{
//			cout<<G[i][j]<<" ";
//		}
//		cout<<endl;
//	}

	dfs(1,0);
	for(int i=1;i<=n;i++)
	{
		//cout<<dp[i]<<endl;
		ans = max(ans,dp[i]); 
	}
	cout<<ans;
	return 0;
}

P1352 没有上司的舞会

思路

树形dp，用 $d p 1 [i]$ 表示以i为根且 $i$ 不参加的子树快乐指数最大值，用 $d p 2 [i]$ 表示以i为根且 $i$ 参加的子树快乐指数最大值。

实现

#include <bits/stdc++.h>
#define maxn 6005
using namespace std;
typedef long long ll;
int ans,n;
int dp1[maxn]; 
int dp2[maxn];
vector<int> G[maxn];
int a[maxn];
bool flag[maxn];
int root;
void dp(int x)
{
	dp2[x] = a[x];
	for(int i=0;i<G[x].size();i++)
	{
		dp(G[x][i]);
		dp1[x] += max(dp1[G[x][i]],dp2[G[x][i]]);
		dp2[x] += dp1[G[x][i]];
	}
}
int main()
{
	int t1,t2;
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
	}
	for(int i=1;i<n;i++)
	{
		scanf("%d%d",&t1,&t2);
		G[t2].push_back(t1);
		flag[t1] = 1;
	}
	for(int i=1;i<=n;i++) 
	{
		if(!flag[i]) root = i;
	}
	dp(root);
	cout<<max(dp1[root],dp2[root]);
	return 0;
}

P2015 二叉苹果树

思路

$d p [i] [j]$ 表示留了 $i$ 号节点和 $i$ 下方的 $j$ 条边时最多可保留的苹果数
枚举 $k$ , $d p [i] [j] = ma x (d p [i] [j], d p [so n] [k] + d p [i] [j - k - 1] + w [i] [so n])$

实现

#include<bits/stdc++.h>
using namespace std;
#define maxn 105
int son[maxn][maxn],f[maxn][maxn];
int n,m,w[maxn][maxn],cnt[maxn],vis[maxn];

void dfs(int k)
{
	vis[k]=1;
	for(int i=1;i<=cnt[k];i++)
	{
		int ny=son[k][i];
		if(vis[ny]==1)continue;
		vis[ny]=1;
		dfs(ny);
		for(int j=m;j>=1;j--) 
			for(int g=j-1;g>=0;g--) 
			{
				f[k][j]=max(f[k][j],f[ny][g]+f[k][j-g-1]+w[k][ny]);
			}
	}
	return;
}

int main()
{
	cin>>n>>m;
	for(int i=1;i<n;i++)
	{
		int x,y,z;
		scanf("%d%d%d",&x,&y,&z);
		w[x][y]=w[y][x]=z;
		son[x][++cnt[x]]=y;
		son[y][++cnt[y]]=x;
	}
	dfs(1);
	cout<<f[1][m]<<endl;
}

P2014 [CTSC1997] 选课

思路

和P2015类似， $d p [i] [j]$ 表示以 $i$ 号节点为根且共保留了包括 $i$ 在内的 $j$ 个节点时最多拥有的学分
枚举 $k$ , $d p [i] [j] = ma x (d p [i] [j], d p [so n] [k] + d p [i] [j - k])$

实现时将0也看成一个节点即可让整个图变成一棵树
需要给m加一

实现

#include<bits/stdc++.h>
using namespace std;
#define maxn 305
vector<int> G[maxn];
int w[maxn];
int n,m;
int f[maxn][maxn];
bool flag[maxn];
void dfs(int k)
{
	for(int i=0;i<G[k].size();i++)
	{
		int ny=G[k][i];
		dfs(ny);
		for(int j=m;j>0;j--) 
			for(int g=1;g<j;g++) 
			{
				f[k][j]=max(f[k][j],f[ny][g]+f[k][j-g]);
			}
	}
	return;
}

int main()
{
	cin>>n>>m;
	for(int i=1;i<=n;i++)
	{
		int x,y,z;
		scanf("%d%d",&x,&y);
		f[i][1]=y;
		G[x].push_back(i);
	}
	++m;
	dfs(0);
	cout<<f[0][m]<<endl;
}

$f [i] [j] [k] = 1$ 表示 $i$ 和 $j$ 之间有一条长度为 $2^k$ 的路径，初始时对于所有连通的 $i$ 和 $j$ ， $f [i] [j] [0] = 1, f [i] [j] [1] = 1$ 当且仅当存在 $k$ 使得 $f [i] [k] [0] == 1$ 且 $f [k] [j] [0] == 1$ 。
最优解路径长度<=maxlongint，所以处理到 $f [i] [j] [64]$ 即可，若 $i$ 和 $j$ 之间有一条长度为 $2^k$ 的路径，则将 $d i s [i] [j]$ 设为1，最后跑一个最短路。

实现

#include<bits/stdc++.h>
using namespace std;
#define maxn 305
const int N=55;
const int K=70;
int f[N][N][K],dis[N][N],n,m;
int main()
{
	int s,t;
    cin>>n>>m;
    memset(dis,0x3f,sizeof dis);
    for(int i=1;i<=m;i++)
    {
    	scanf("%d%d",&s,&t);
        f[s][t][0]=1;
        dis[s][t]=1;
    }
    
    for(int o=1;o<=64;o++)
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                for(int k=1;k<=n;k++)
                    if(f[j][i][o-1]==true and f[i][k][o-1]==true)
                    {
                        f[j][k][o]=true;
                        dis[j][k]=1;
                    }
                
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            for(int k=1;k<=n;k++)
                dis[j][k]=min(dis[j][k],dis[j][i]+dis[i][k]);
    
    printf("%d",dis[1][n]);
    return 0;
}

P1040 [NOIP2003 提高组] 加分二叉树

思路

$f [i] [j]$ 为节点 $i$ 到 $j$ 的最大得分，答案即为 $f [1] [n]$ 。对于二叉树的中序遍历，任意选定一个点，左边的就是左子树，右边的就是右子树，可以很方便的做区间dp。
枚举 $k$ :
$f [i] [j] = f [i] [k - 1] * f [k + 1] [j] + f [k] [k]$
使得 $f [i] [j]$ 最大的 $k$ 值保存下来， $roo t [i] [j] = k$
递归输出区间[l,r]的前序序列：
$roo t [l, r]$ ，区间 $[l, roo t [l, r] - 1]$ 的前序序列，区间 $[roo t [l, r] + 1, r]$ 的前序序列

实现

来自洛谷题解区

#include<iostream>
#include<cstdio>
using namespace std;
int n,v[39],f[47][47],i,j,k,root[49][49];
void print(int l,int r){
    if(l>r)return;
    if(l==r){printf("%d ",l);return;}
    printf("%d ",root[l][r]);
    print(l,root[l][r]-1);
    print(root[l][r]+1,r);
}
int main() {
    scanf("%d",&n);
    for( i=1; i<=n; i++) scanf("%d",&v[i]);
    for(i=1; i<=n; i++) {f[i][i]=v[i];f[i][i-1]=1;}
    for(i=n; i>=1; i--)
        for(j=i+1; j<=n; j++)
            for(k=i; k<=j; k++) {
                if(f[i][j]<(f[i][k-1]*f[k+1][j]+f[k][k])) {
                    f[i][j]=f[i][k-1]*f[k+1][j]+f[k][k];
                    root[i][j]=k;
                }
            }
    printf("%d\n",f[1][n]);
    print(1,n);
    return 0;
}