【树状数组】PKU-2152-Fire

最新推荐文章于 2022-03-21 21:03:11 发布

原创最新推荐文章于 2022-03-21 21:03:11 发布 · 544 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#struct #ini

动态规划（DP）专栏收录该内容

13 篇文章

订阅专栏

本文介绍了一种使用树状数组优化的深度优先搜索（DFS）算法解决特定问题的方法。该算法通过预先计算节点间距离并限制最远可达距离来减少计算复杂度，适用于解决涉及树形结构中路径和最小化问题。

一道处理的很巧妙的树状数组……都是参考大牛的代码的……

题目

大牛题解

#include<iostream>
#include<vector>
using namespace std;
template<class T> T Max(T x,T y){return x>y?x:y;}
template<class T> T Min(T x,T y){return x<y?x:y;}
#define N 1005
#define inf 999999999
struct node
{
	int u,w;
	node(int x,int y)
	{
		u=x,w=y;
	}
};
vector<node> v[N];
int n,d[N],w[N],dist[N],best[N],dp[N][N];
void dis(int u)
{
	int i,x,len;
	len=v[u].size();
	for(i=0;i<len;i++)
	{
		x=v[u][i].u;
		if(dist[x]!=-1)continue;
		dist[x]=dist[u]+v[u][i].w;
		dis(x);
	}
}
void dfs(int now,int pre)
{
	int i,j,u,len;
	len=v[now].size();
	for(i=0;i<len;i++)
	{
		u=v[now][i].u;
		if(u==pre)continue;
		dfs(u,now);
	}
	memset(dist,-1,sizeof(dist));
	dist[now]=0;
	dis(now);
	best[now]=inf;
	for(i=1;i<=n;i++)
	{
		if(dist[i]>d[now])
		{
			dp[now][i]=inf;
			continue;
		}
		dp[now][i]=w[i];
		for(j=0;j<len;j++)
		{
			u=v[now][j].u;
			if(u==pre)continue;
			dp[now][i]+=Min(best[u],dp[u][i]-w[i]);
		}
		best[now]=Min(best[now],dp[now][i]);
	}
}
int main()
{
	//freopen("a.txt","r",stdin);
	int t,i,x,y,z;
	scanf("%d",&t);
	while(t--)
	{
		scanf("%d",&n);
		for(i=1;i<=n;i++)scanf("%d",w+i);
		for(i=1;i<=n;i++)
		{
			scanf("%d",d+i);
			v[i].clear();
		}
		for(i=1;i<n;i++)
		{
			scanf("%d%d%d",&x,&y,&z);
			v[x].push_back(node(y,z));
			v[y].push_back(node(x,z));
		}
		dfs(1,0);
		printf("%d\n",best[1]);
	}
	return 0;
}