Problem 2169 shadow
Accept: 425 Submit: 1666
Time Limit: 1000 mSec Memory Limit : 32768 KB
Problem Description
YL是shadow国的国王,shadow国有N个城市。为了节省开支,shadow国只有N-1条道路,这N-1条道路使得N个城市连通。某一年,shadow国发生了叛乱,叛军占领了多个城市,王都岌岌可危。王都为编号为1的城市,除了王都外有K个城市有YL的军队。现在这K支军队要向王都进军,并且消灭沿途经过的城市中的叛军。现给出N个城市的道路情况以及城市的叛军数量,问总共需要消灭多少叛军?
Input
第一行输入两个整数N,K,接下来输入N(1<=N<=100000)个整数Ai(0<=Ai<=10000),表示第i个城市的叛军数量。接下来输入K个大于等于1且小于等于N的整数,表示有军队的城市的编号。数据保证王都以及有军队的城市没有叛军。接下来输入N-1行,每行两个整数u、v,表示连接u和v的一条道路。每支军队只能沿着道路走,并且是其所在城市与王都之间的最短路线走。
Output
输出一行一个整数表示消灭的叛军数量。
Sample Input
4 2 0 3 0 0 3 4 1 2 2 3 2 4
Sample Output
3
虽然数据很弱,可以很简单的代码就过,但是还是不要错过这个简单的图算法题目。
广度优先搜索,从1开始搜遇到一个有友军的城市就回溯,将路径上的所有敌军消灭,一个队列就可以解决,但是需要制作很多标记,敌军标记,友军标记,访问标记,然后队列就要有上一个城市标记,图因为太大使用邻接表存储,ylarmy.a[i]为0表示没有访问过,1表示访问过,2表示友军所在城市,为3表示皇城。ylarmy.b[i]表示回溯过的城市,也就是走过的路,这个标记是防止走重复路的,因为每一次回溯都回溯到皇城的话就会走太多的重复路,差不多就是dp中剪枝的意思。代码如下:
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
using namespace std;
const int maxn=1e5+5;
struct Enode{int endpoint;Enode *nextEdg;};//边终点标记
struct Vnode{int reble_army;Enode *firstedg;};//叛军//第一条边
struct Graph{Vnode vnode[maxn];int n;}g;
struct{int a[maxn];bool b[maxn];}ylarmy;
struct queue{int pre,d;};
void clearg(int n){
g.n=n;
for(int i=0;i<=n;i++)
g.vnode[i].firstedg=NULL;
}
void insert(int start,int end){
Enode *p=(Enode*)malloc(sizeof(Enode));
p->endpoint=end;
p->nextEdg=g.vnode[start].firstedg;
g.vnode[start].firstedg=p;
}
int Bfs(){
int ans=0,x;Enode *e;
queue q[maxn];int front=-1,rear=-1;
rear++;q[rear].pre=-1;q[rear].d=1;
ylarmy.a[1]=3;ylarmy.b[1]=true;
while(front!=rear){
e=g.vnode[q[++front].d].firstedg;
while(e!=NULL{
if(ylarmy.a[e->endpoint]==2){
ans+=g.vnode[e->endpoint].reble_army;
int d=q[front].d;
int pre=q[front].pre;
while(!ylarmy.b[d]){
ans+=g.vnode[d].reble_army;
g.vnode[d].reble_army=0;
ylarmy.b[d]=true;
d=q[pre].d;
pre=q[pre].pre;
}
ylarmy.a[e->endpoint]=0;
}
if(ylarmy.a[e->endpoint]==0){
ylarmy.a[e->endpoint]=1;///表示已经走过的点
q[++rear].pre=front;
q[rear].d=e->endpoint;
}
e=e->nextEdg;
}
}
return ans;
}
int main()
{
int n,k;
int i,p1,p2,d;
while(~scanf("%d%d",&n,&k)){
memset(ylarmy.a,0,sizeof(ylarmy.a));
memset(ylarmy.b,0,sizeof(ylarmy.b));
clearg(n);
for(i=1;i<=n;i++)
scanf("%d",&g.vnode[i].reble_army);
for(i=0;i<k;i++){
scanf("%d",&d);
ylarmy.a[d]=2;///表示友军
}
i=n-1;
while(i--){
scanf("%d%d",&p1,&p2);
insert(p1,p2);
insert(p2,p1);
}
printf("%d\n",Bfs());
}
return 0;
}
有点长,但是思路很清晰,容易理解。