思想:这道题把目标节点当做一棵树的根节点,然后2条边同时遍历提取信息,做dfs,把信息放在对应边的节点上而不是边上,是一道非常有启发的题!N如果作为全局变量,给全局数组初始化会导致错误。
这道题的核心思想是如何存取生成一个结点的2个结点的边。如果一个结点,例如,1号结点被2,3号结点生成且被2,4号结点生成。此时,就会出现重边。巧妙的是,不像传统的邻接表遍历方式,此问题每次遍历邻接表的2条边,从存取顺序来说,一定会正确的取到生成此结点的2个结点关系。
#include<iostream>
#include<vector>
#include<climits>
using namespace std;
int N,M,K,Ta;
const int cap=2e5+10;//边是二倍
vector<int> h;//头数组里面记录的是边
int e[cap];//当前边的上一条边
int v[cap];
int num=0;
vector<int> ans(2010,INT_MAX);
vector<int> T;
void add(int a,int b)
{
e[num]=h[a];
v[num]=b;
h[a]=num++;//记录的边号
}
int dfs(int node)
{
if(ans[node]!=INT_MAX)
return ans[node];
for(int i=h[node];i!=-1;i=e[e[i]])//i代表编号
{
int j=e[i];
ans[node]=min(ans[node],max(T[v[i]],T[v[j]])+max(dfs(v[i]),dfs(v[j])));
}
return ans[node];
}
int main()
{
ios::sync_with_stdio(0);
cin.tie(0);
cin>>N>>M>>K>>Ta;
T.resize(N+1);
for(int i=1;i<=N;i++)
cin>>T[i];
h.resize(N+1,-1);
vector<bool> mark(N+1,0);
for(int i=0;i<M;i++)
{
int w;cin>>w;
ans[w]=0;
}
for(int i=0;i<K;i++)
{
int a,b,c;
cin>>a>>b>>c;
add(c,a);
add(c,b);
}
cout<<dfs(Ta)<<'\n';
return 0;
}