hiho1063(树型dp)

qkoqhh

于 2018-03-29 21:47:26 发布

阅读量138

点赞数

分类专栏： DP

本文链接：https://blog.csdn.net/qkoqhh/article/details/79749670

版权

DP 专栏收录该内容

116 篇文章 1 订阅

订阅专栏

有点意思这题。。首先要在状态上选好。。看到距离是10^6就放弃将距离放入状态的念头，那么距离就只能放在数组里面了吧。。所以剩下的就是状态就是点和价值了，所以答案反而是在数组下标里面找。。

然后转移呢。。考虑到可能有回来和不回来的情况多来一维，然后设d[i][j][k]为在第i点为根的子树要获得j价值的最小距离，k==0表示回来,k==1表示不回来

然后先看k==0的情况，对一个确定的i进行转移，其实就是做一个小背包啦。。

对k==1的情况，对每个儿子要么选它不回来，或者选已经选的不回来的状态加上它回来，然后同样是个小背包。。

语文不好。。看代码领悟qaq

/**
 *　　　　　　　　┏┓　　 　┏┓
 * 　　　　　　　┏┛┗━━━━━━━┛┗━━━┓
 * 　　　　　　　┃　　　　　　　┃ 　
 * 　　　　　　　┃　　　━　　 　┃
 * 　　　　　　　┃　＞　　　＜　┃
 * 　　　　　　　┃　　　　　　　┃
 * 　　　　　　　┃...　⌒　... 　┃
 * 　　　　　　　┃　　　　　　　┃
 * 　　　　　　　┗━┓　　　┏━┛
 * 　　　　　　　　　┃　　　┃　Code is far away from bug with the animal protecting　　　　　　　　　　
 * 　　　　　　　　　┃　　　┃   神兽保佑,代码无bug
 * 　　　　　　　　　┃　　　┃　　　　　　　　　　　
 * 　　　　　　　　　┃　　　┃  　　　　　　
 * 　　　　　　　　　┃　　　┃
 * 　　　　　　　　　┃　　　┃　　　　　　　　　　　
 * 　　　　　　　　　┃　　　┗━━━┓
 * 　　　　　　　　　┃　　　　　　　┣┓
 * 　　　　　　　　　┃　　　　　　　┏┛
 * 　　　　　　　　　┗┓┓┏━┳┓┏┛
 * 　　　　　　　　　　┃┫┫　┃┫┫
 * 　　　　　　　　　　┗┻┛　┗┻┛
 */ 
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<queue>
#include<cmath>
#include<map>
#include<stack>
#define inc(i,l,r) for(int i=l;i<=r;i++)
#define dec(i,l,r) for(int i=l;i>=r;i--)
#define link(x) for(edge *j=h[x];j;j=j->next)
#define mem(a) memset(a,0,sizeof(a))
#define ll long long
#define eps 1e-12
#define succ(x) (1<<x)
#define lowbit(x) (x&(-x))
#define sqr(x) ((x)*(x))
#define mid (x+y>>1)
#define NM 205
#define nm 1000005
#define pi 3.1415926535897931
using namespace std;
const int inf=1000000007;
ll read(){
    ll x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
    while(isdigit(ch))x=x*10+ch-'0',ch=getchar();
    return f*x;
}




struct ttmp{
	int x,y;
	bool operator<(const ttmp&o)const{return y<o.y||(y==o.y&&x<o.x);}
}b[2*NM];
struct edge{int t,v;edge*next;}e[nm],*h[NM],*o=e;
void add(int x,int y,int v){o->t=y;o->v=v;o->next=h[x];h[x]=o++;}
int n,m,a[NM],d[NM][NM][2],_x,_y,_t,f[NM],tot,tmp[nm],c[nm],ans[nm];
void dfs(int x){
	d[x][a[x]][0]=d[x][a[x]][1]=0;
	link(x)if(!f[j->t]){
		f[j->t]=x;dfs(j->t);
		dec(i,tot,a[x]+1)
			inc(k,1,i-a[x])d[x][i][0]=min(d[x][i][0],d[j->t][k][0]+d[x][i-k][0]+2*j->v),
			d[x][i][1]=min({d[x][i][1],d[x][i-k][0]+d[j->t][k][1]+j->v,d[x][i-k][1]+d[j->t][k][0]+2*j->v});
	}
}

bool cmp(int x,int y){return c[x]<c[y];}
int main(){
//	freopen("data.in","r",stdin);
	n=read();
	inc(i,1,n)a[i]=read();
	inc(i,1,n-1){_x=read();_y=read();_t=read();add(_x,_y,_t);add(_y,_x,_t);}
	tot=n<<1;
	inc(i,1,n)inc(j,0,tot)inc(k,0,1)d[i][j][k]=inf;
	dfs(f[1]=1);
	m=read();
	inc(i,0,tot)b[i].y=d[1][i][1],b[i].x=i;
	//inc(i,0,tot)printf("%d ",b[i].y);putchar('\n');
	//inc(i,0,tot)printf("%d ",b[i].x);putchar('\n');
	sort(b,b+1+tot);
	inc(i,1,m)c[i]=read(),tmp[i]=i;
	sort(tmp+1,tmp+1+m,cmp);_x=0;_y=0;
	for(int k=1,i=tmp[k];k<=m;i=tmp[++k]){
		while(_x<=tot&&b[_x].y<=c[i])_y=max(_y,b[_x++].x);
		ans[i]=_y;
	}
	inc(i,1,m)printf("%d\n",ans[i]);
	return 0;
}