GDUT——1169: Krito的讨伐（优先队列BFS）

1169: Krito的讨伐

Time Limit: 1 Sec Memory Limit: 128 MB
Submit: 619 Solved: 102

Description

Krito终于干掉了99层的boss，来到了第100层。第100层可以表示成一颗树，这棵树有n个节点（编号从0到n-1），树上每一个节点可能有很多只怪物。 Krito现在在0号节点，现在它想要区清除这一层所有的怪物。他现在有atk大小的攻击力。只有当你的攻击力大于这只怪物的防御力时，你才可以打败他，同时每打败只怪物,你会获得一定的攻击力加成。一个节点可能存在着不止一只怪兽，你要打败这个节点的所有怪物才能可以从这个节点通过，请问他能不能完成这个任务？注意：不要求一次性杀光一个节点里面的所有怪物。

Input

第1行：一个数T，表示有T个测试样例（0<=T<=50) ,接下来有T个测试样例

对于每一个测试样例：

第1行：两个整数n，m表示这棵树有n个节点，m只怪兽(0<=n<=1000 ,0<=m <=100)

第2至n-1行：两个整数u，v表示编号为u，v之间的节点有一条无向边，题目保证不会成环。（0<=u,v<n , u!=v)

第3行：一个整数atk，表示Krito的初始化攻击力(0<=atk<=100)

第4至3+m行：两个整数id，def，add_atk，表示在编号为id的点上，有一只防御力为def的怪物,打败后可以增加add_atk点的攻击力。(0<=add_atk,def<=100)

Output

对于每一个测试样例，如果Krito能够清除所有的怪物，则输出“Oh yes.” 否则，输出“Good Good Study,Day Day Up.”

Sample Input

15 20 10 22 32 4113 10 21 11 0

Sample Output

Oh yes.

一直觉得这题有毒，用最后m是否等于0的方法写死活WA，然后换个方法重新写了下用从0开始拓展的方法写就可以过。

先上正确的代码：

#include<iostream>

#include<algorithm>

#include<cstdlib>

#include<sstream>

#include<cstring>

#include<cstdio>

#include<string>

#include<deque>

#include<stack>

#include<cmath>

#include<queue>

#include<set>

#include<map>

using namespace std;

#define INF 0x3f3f3f3f

#define MM(x,y) memset(x,y,sizeof(x))

typedef pair<int,int> pii;

typedef long long LL;

const double PI=acos(-1.0);

const int M=2010;

const int N=1010;

struct info

{

	int to;

	int pre;

};

struct mons

{

	int d;

	int add;

	int cur;

	bool operator<(const mons &b)const

	{

		return d>b.d;

	}

};

info E[M];

int head[N],cnt,ran[N];

int vis[N],n,m,atk;

vector<mons> guaiwu[N];

priority_queue<mons>Q;

void add(int s,int t)

{

	E[cnt].to=t;

	E[cnt].pre=head[s];

	head[s]=cnt++;

}

void init()

{

	MM(head,-1);

	MM(vis,0);

	MM(ran,0);

	MM(vis,0);

	cnt=0;

	for (int i=0; i<N; i++)

		guaiwu[i].clear();

	while (!Q.empty())

		Q.pop();

}

void findmons(const int &s)

{

	if(guaiwu[s].empty())

		return ;

	for (int i=0; i<guaiwu[s].size(); i++)

		Q.push(guaiwu[s][i]);

	guaiwu[s].clear();

}

void dfs(const int &now)

{

	for (int i=head[now]; ~i; i=E[i].pre)

	{

		int v=E[i].to;

		if(!vis[v])

		{

			vis[v]=1;

			if(!ran[v])

				dfs(v);

			else

				findmons(v);

		}

	}

}

int kill(const int &s)

{

	if(!ran[s])

		dfs(s);

	else

		findmons(s);

	while (!Q.empty())

	{

		mons now=Q.top();

		Q.pop();

		if(now.d>=atk)

			return 0;

		else

		{

			atk+=now.add;

			ran[now.cur]--;

			if(!ran[now.cur])

				dfs(now.cur);

		}

	}

	return 1;

}

int main(void)

{

	int tcase,i,j,a,b,D,A,X;

	scanf("%d",&tcase);

	while (tcase--)

	{

		init();

		scanf("%d%d",&n,&m);

		for (i=0; i<n-1; ++i)

		{

			scanf("%d%d",&a,&b);

			add(a,b);

			add(b,a);

		}

		scanf("%d",&atk);

		mons t;

		for (i=0; i<m; i++)

		{

			scanf("%d%d%d",&t.cur,&t.d,&t.add);

			guaiwu[t.cur].push_back(t);

			ran[t.cur]++;

		}

		puts(!kill(0)?"Good Good Study,Day Day Up.":"Oh yes.");

	}

	return 0;

}

修改N次仍然是WA的代码（不解题目既然保证了是一棵树不知道错的哪里）：

#include<iostream>

#include<algorithm>

#include<cstdlib>

#include<sstream>

#include<cstring>

#include<cstdio>

#include<string>

#include<deque>

#include<stack>

#include<cmath>

#include<queue>

#include<set>

#include<map>

using namespace std;

#define INF 0x3f3f3f3f

#define MM(x,y) memset(x,y,sizeof(x))

typedef pair<int,int> pii;

typedef long long LL;

const double PI=acos(-1.0);

const int N=1010;

struct info

{

	int cur;

	int d;

	int ATK;

	info(int c,int dd,int aa):cur(c),d(dd),ATK(aa){}

	info(){}

	bool operator<(const info &b)const

	{

		return d>b.d;

	}

};

vector<info>mon[N];

vector<int>E[N];

priority_queue<info>kill;

int ran[N];

int vis[N];

int n,m,atk;

void reachble(const int &s)

{

	int i,j;

	queue<int>Q;

	Q.push(s);

	while (!Q.empty())

	{

		int now=Q.front();

		Q.pop();

		for (i=0; i<E[now].size(); i++)

		{

			int v=E[now][i];

			if(!vis[v])

			{

				vis[v]=1;

				if(!ran[v])

					Q.push(v);

				else

				{

					for (j=0; j<mon[v].size(); j++)

					{

						if(mon[v][j].d!=-1)

						{

							kill.push(info(v,mon[v][j].d,mon[v][j].ATK));

							mon[v][j].d=-1;

						}

					}

				}

			}

		}

	}

}

void init()

{

	for (int i=0; i<N; i++)

	{

		E[i].clear();

		mon[i].clear();

	}

	while (!kill.empty())

		kill.pop();

	MM(vis,0);

	MM(ran,0);

}

bool cmp(const info &a,const info &b)

{

	return a.d<b.d;

}

int main(void)

{

	int i,j,tcase,x,y,d,a;

	scanf("%d",&tcase);

	for (int t=1; t<=tcase; t++)

	{

		init();

		scanf("%d%d",&n,&m);

		for (i=1; i<n; i++)

		{

			scanf("%d%d",&x,&y);

			E[x].push_back(y);

			E[y].push_back(x);

		}

		scanf("%d",&atk);

		for (i=0; i<m; i++)

		{

			scanf("%d%d%d",&x,&d,&a);

			mon[x].push_back(info(x,d,a));

			ran[x]++;

		}

		int flag=1;

		sort(mon[0].begin(),mon[0].end(),cmp);

		for (i=0; i<mon[0].size(); i++)

		{

			if(mon[0][i].d>=atk)

			{

				flag=0;

				break;

			}

			else

			{

				atk+=mon[0][i].ATK;

				--m;

				--ran[0];

			}

		}

		if(!flag)

		{

			puts("Good Good Study,Day Day Up.");

			continue;

		}

		vis[0]=1;

		reachble(0);

		while (!kill.empty())

		{

			info now=kill.top();

			kill.pop();

			if(now.d>=atk)

				break;

			atk+=now.ATK;

			--m;

			if(--ran[now.cur]==0)

				reachble(now.cur);

		}

		puts(m?"Good Good Study,Day Day Up.":"Oh yes.");

	}

	return 0;

}

巴特西