首页 > 代码库 > ACdream1032 Component
ACdream1032 Component
题解:
树形dp+背包
不难想到,dp[i[j]表示以i为根的子树中,有j个子树的最小和
Size数组记录子树节点个数,然后就树上背包了
树上背包写时,在更新时,需要逆写
这里不懂,问了下q巨,QRZ
代码:
#include<iostream>#include<cstdio>#include<vector>#include<cstring>#include<cmath>#include<queue>using namespace std;#define pb push_back#define mp make_pair#define se second#define fs first#define ll long long#define CLR(x) memset(x,0,sizeof x)#define MC(x,y) memcpy(x,y,sizeof(x)) #define SZ(x) ((int)(x).size())#define FOR(it,c) for(__typeof((c).begin()) it=(c).begin();it!=(c).end();it++)#define lson l,m,rt<<1#define rson m+1,r,rt<<1|1 typedef pair<int,int> P;const double eps=1e-9;const int maxn=2050;const int mod=1e9+7;const ll INF=1e18; ll dp[maxn][maxn],ans[maxn];int Size[maxn],p[maxn],head[maxn];int n,cnt; struct Edge{ int v,nxt;}edge[maxn*2]; void Init(){ cnt=0; memset(Size,0,sizeof(Size)); memset(head,-1,sizeof(head)); for(int i=1;i<=n;i++) for(int j=1;j<=n;j++) dp[i][j]=INF; for(int i=1;i<=n;i++) ans[i]=INF;} void AddEdge(int u,int v){ edge[cnt].v=v; edge[cnt].nxt=head[u]; head[u]=cnt++;} void dfs(int u,int fa){ dp[u][1]=p[u]*1LL; Size[u]=1; for(int i=head[u];i!=-1;i=edge[i].nxt){ int v=edge[i].v; if(v==fa) continue; dfs(v,u); Size[u]+=Size[v]; for(int j=Size[u];j>=2;j--) for(int k=min(Size[v],j-1);k>=1;k--) dp[u][j]=min(dp[u][j],dp[u][j-k]+dp[v][k]); }} int main(){ int u,v; scanf("%d",&n); Init(); for(int i=1;i<=n;i++) scanf("%d",&p[i]); for(int i=1;i<=n-1;i++){ scanf("%d%d",&u,&v); AddEdge(u,v);AddEdge(v,u); } dfs(1,-1); for(int i=1;i<=n;i++) for(int j=1;j<=n;j++) ans[j]=min(ans[j],dp[i][j]); for(int i=1;i<=n;i++) printf("%lld%c",ans[i],i==n?‘\n‘:‘ ‘); return 0;}
ACdream1032 Component
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。