FZU2169 shadow题解

#include<iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include<vector>
#include<queue>
#include<set>
#define MAXN 101111
#define RE freopen("in.txt","r",stdin);

using namespace std;

struct Edge
{
int u,v;
int next;
};
Edge e[MAXN*2];
int first[MAXN];
int en;

void add(int u,int v)
{
e[en].u=u;
e[en].v=v;
e[en].next=first[u];
first[u]=en;
en++;
}

int n,k,cnt;
int a[MAXN];
set<int> b;
bool walked[MAXN];
bool geted[MAXN];
queue<int> v;
int from[MAXN];
int farm();
void init();

int main()
{
//RE
while(scanf("%d%d",&n,&k)!=EOF)
{
farm();
}
return 0;
}

int farm()
{
int i,j,x,y;
init();
for(i=0; i<n; i++)
scanf("%d",&a[i]);
for(i=0; i<k; i++)
{
scanf("%d",&x);
x--;
b.insert(x);
}
for(i=0; i<n-1; i++)
{
scanf("%d%d",&x,&y);
x--;y--;
add(y,x);
add(x,y);
}
long long ans=0;
while(!v.empty()) v.pop();
v.push(0);
walked[0]==true;
geted[0]=true;
while(!v.empty())
{
int now=v.front();
v.pop();
//cout<<now<<'!';
if(b.find(now)!=b.end())
{
//cout<<now<<',';
x=now;
while(!geted[x])
{
geted[x]=true;
ans+=a[x];
x=from[x];
}
cnt++;
if(cnt==k) break;
continue;
}
for(j=first[now]; j!=-1; j=e[j].next)
{
//cout<<j<<'?';
int next=e[j].v;
if(!walked[next])
{
walked[next]=true;
v.push(next);
from[next]=now;
}
}
}
printf("%lld\n",ans);
return 0;
}

void init()
{
memset(e,0,sizeof(e));
memset(first,-1,sizeof(first));
memset(walked,false,sizeof(walked));
memset(geted,false,sizeof(geted));
b.clear();
cnt=0;
en=0;
}

内容来自用户分享和网络整理，不保证内容的准确性，如有侵权内容，可联系管理员处理

• 添加评论
• 分享网址
• 分享文章
• 返回顶部