P3691树上涂色
时间限制 : - MS 空间限制 : 65536 KB
评测说明 : 时限1000ms
问题描述
给出一棵有n个节点(编号1到n)的树,开始时每个节点都没有颜色。现在有A和B两种颜色,给第x号节点涂上A颜色需要花费Ax块钱,给x号节点涂上B颜色需要花费Bx块钱。
如果i,j两个节点相邻,也就是i和j之间有边直接相连。若i号点已经涂成了A色,那么给j号点涂A色只需Aj/2块钱。若i号点已经涂成了B色,那么给j号点涂B色只需Bj/2块钱。也就是说,如果相邻点已经被涂了相同的颜色,当前点涂这种颜色的时候只需要一半的费用。
现在需要将树上所有点都涂上颜色,问最少需要多少费用?
如果i,j两个节点相邻,也就是i和j之间有边直接相连。若i号点已经涂成了A色,那么给j号点涂A色只需Aj/2块钱。若i号点已经涂成了B色,那么给j号点涂B色只需Bj/2块钱。也就是说,如果相邻点已经被涂了相同的颜色,当前点涂这种颜色的时候只需要一半的费用。
现在需要将树上所有点都涂上颜色,问最少需要多少费用?
输入格式
第一行,一个整数n,表示节点总数
第二行,n个空格间隔的整数,分别表示给1到n号节点涂上A颜色所需费用
第三行,n个空格间隔的整数,分别表示给1到n号节点涂上B颜色所需费用
接下来n-1行,每行两个整数i和j,表示i和j节点间有边直接相连
输出格式
一行,一个整数,所需最小费用
样例输入 1
3
1 2 5
3 8 1
1 2
1 3
样例输出 1
3
样例输入 2
6
1 40 2 5 6 10
4 14 4 2 8 30
1 2
1 3
2 4
4 5
4 6
样例输出 2
25
提示
1<=n<=100
0<=Ax,Bx<=10000
分析:
状态:
f[i][0]: 以i为根的子树,涂上颜色的最小代价。其中i涂A颜色,与i连通的同色块(在子树内的)中,所有点都是以半价涂色的(即没有“涂色起点”)。
f[i][1]: 以i为根的子树,涂上颜色的最小代价。其中,i涂A颜色,与i连通的同色块中,存在一个节点是以全部代价涂色的(即有“涂色起点”)。
g[i][0]: 以i为根的子树,涂上颜色的最小代价。其中i涂B颜色,与i连通的同色块中,所有点都是以半价涂色的。
g[i][1]: 以i为根的子树,涂上颜色的最小代价。其中i涂B颜色,与i连通的同色块中,存在一个节点是以全部代价涂色
f[i][0]=Sum{ min( f[ j][0] , g[ j][1] ) } ( j为i的所有儿子节点)
1.为什么不讨论f[j][1]呢?
2.为什么不讨论g[j][0]呢?
想一想,为什么?
方程:
f[i][0]=Sum{ min( f[ j][0] , g[ j][1] ) } ( j为i的所有儿子节点)
f[i][1]=min{
f[i][0]+ A[i]/2 //i涂全价A色
f[i][0]+ ChaJia //i的一个同色连通子树中存在一个节点涂全价A色
}
ChaJia=min{ f[j][1]-min(f[j][0] , g[j][1] ) } j为i的所有儿子节点
说明:
f[i][1]表示i为根的字数中,i涂A色,且与i同色的连通块中,有一个点是全价A的最优方案。 我们可以对i涂全价A,也可以使与A同色连通的某棵子树中存在全价A的点。
而f[i][0]表示i为根的子树中,i涂A色,且与i同色的连通块全是半价A的最优方案。求f[i][1]只需在f[i][0]上做调整
如果我们直接将i涂成全价的A,即是f[i][0]+A[i]/2
如果我们不把i涂成全价的A,而是选一棵与i同色连通的且存在全价A的子树替换掉原来的子树,即是f[i][0]+ChaJia 这棵子树原来加入f[i][0]的状态是min(g[j][1],f[j][0]),而新替换进的状态是f[j][1],两者之差即为替换产生的差价。
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#define inf 1e9
using namespace std;
struct line{
int from,to;
};
line edge[1000005];
int last[1005],_next[100005];
int n,m=0;
int a[105],b[105];
int f[105][5],g[105][5];
void add_edge(int x,int y){
m++;
_next[m]=last[x];
last[x]=m;
edge[m].from=x;
edge[m].to=y;
}
void dp(int x,int fa){
int i,j,k;
bool leaf=true;
int cha1=inf,cha2=inf;
/*if(last[x]==0){
f[x][1]=a[x];g[x][1]=b[x];
f[x][0]=a[x]/2;g[x][0]=b[x]/2;
return;
}*/
for(j=last[x];j;j=_next[j]){
i=edge[j].to;
if(i==fa)continue;
dp(i,x);
leaf=false;
f[x][0]+=min(f[i][0],g[i][1]);
g[x][0]+=min(f[i][1],g[i][0]);
cha1=min(cha1,f[i][1]-min(f[i][0],g[i][1]));
cha2=min(cha2,g[i][1]-min(f[i][1],g[i][0]));
}
if(leaf){
f[x][1]=a[x];g[x][1]=b[x];
f[x][0]=a[x]/2;g[x][0]=b[x]/2;
}
else{
f[x][0]+=a[x]/2;
g[x][0]+=b[x]/2;
f[x][1]=f[x][0]+min(a[x]/2,cha1);
g[x][1]=g[x][0]+min(b[x]/2,cha2);
}
}
int main(){
int i,j,k;
cin>>n;
for(i=1;i<=n;i++){
scanf("%d",&a[i]);
}
for(i=1;i<=n;i++){
scanf("%d",&b[i]);
}
for(i=1;i<n;i++){
int x,y;
scanf("%d%d",&x,&y);
add_edge(x,y);
add_edge(y,x);
}
dp(1,0);
cout<<min(f[1][1],g[1][1]);
}
/*
f[i][0]表示i涂a并且连通的全半价
f[i][1]表示i涂a并且连通的1个全价
g[i][0]表示i涂b并且连通的全半价
g[i][1]表示i涂b并且连通的1个全价
*/