poj 2152 Fire

06年的论文题（终于刷到06年了）

陈启峰神犇讲的

Discuss里面有人说他那个复杂度是n*n*sigma(deg[i])的

其实我想说明明是n*sigma(deg[i])=n*n好吗

然而不要问我为什么我的程序跑了1700多MS（捂脸，蒟蒻自带1000的常数）

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
using namespace std;
const int N=1000+5;
const int inf=1e9;
struct Edge{int to,next,v;}e[N<<1];
int w[N],d[N],dis[N],dep[N],st[N<<1][13],sz,pos[N],l[N],r[N];
int head[N],cnt;
void ins(int u,int v,int w){
	e[++cnt]=(Edge){v,head[u],w};head[u]=cnt;
}
void dfs(int u,int fa,int depth){
	dep[u]=depth;st[++sz][0]=u;pos[u]=sz;l[u]=sz;
	for(int i=head[u];i;i=e[i].next)
	if(e[i].to!=fa){
		dis[e[i].to]=dis[u]+e[i].v;
		dfs(e[i].to,u,depth+1);
		st[++sz][0]=u;
	}
	r[u]=sz;
}
bool cmp(int i,int j){
	return dep[i]<dep[j];
}
void rmq_init(){
	for(int j=1;1<<j<=sz;j++)
	for(int i=1;i+(1<<j)-1<=sz;i++)
	st[i][j]=min(st[i][j-1],st[i+(1<<j-1)][j-1],cmp);
}
int lca(int u,int v){
	u=pos[u];v=pos[v];if(u>v)swap(u,v);
	int k=log2(v-u+1);
	return min(st[u][k],st[v-(1<<k)+1][k],cmp);
}
int dist(int u,int v){
	return dis[u]+dis[v]-2*dis[lca(u,v)];
}
bool in(int u,int v){
	return l[u]<=pos[v]&&pos[v]<=r[u];
}
int f[N][N],best[N],n;
void dp(int u,int fa){
	for(int i=1;i<=n;i++)f[u][i]=inf;
	best[u]=inf;
	for(int i=head[u];i;i=e[i].next)
	if(e[i].to!=fa)dp(e[i].to,u);
	for(int v=1;v<=n;v++)
	if(dist(u,v)<=d[u]){
		int sum=0;
		for(int i=head[u];i;i=e[i].next)
		if(e[i].to!=fa)
		sum+=min(best[e[i].to],f[e[i].to][v]);
		if(u==v)f[u][v]=w[v]+sum;
		else if(in(u,v))
		for(int i=head[u];i;i=e[i].next){
		if(e[i].to!=fa&&in(e[i].to,v)){
			f[u][v]=f[e[i].to][v]+sum-min(best[e[i].to],f[e[i].to][v]);
			break;
		}}
		else f[u][v]=sum;
	}
	for(int v=1;v<=n;v++)
	if(in(u,v))
	best[u]=min(best[u],f[u][v]);
}
int main(){
	//freopen("a.in","r",stdin);
	int T;scanf("%d",&T);
	while(T--){
		scanf("%d",&n);
		for(int i=1;i<=n;i++)scanf("%d",&w[i]);
		for(int i=1;i<=n;i++)scanf("%d",&d[i]);
		memset(head,0,sizeof(head));cnt=0;
		int u,v,w;
		for(int i=1;i<n;i++){
			scanf("%d%d%d",&u,&v,&w);
			ins(u,v,w);ins(v,u,w);
		}
		sz=0;dfs(1,-1,0);rmq_init();dp(1,-1);
		printf("%d\n",best[1]);
	}
	return 0;
}