【BZOJ3417】Poi2013 Tales of seafaring 分层图BFS

【BZOJ3417】Poi2013 Tales of seafaring

Description

一个n点m边无向图，边权均为1，有k个询问

每次询问给出(s,t,d)，要求回答是否存在一条从s到t的路径，长度为d

路径不必是简单路(可以自交)

2<=N<=5000,1<=M<=5000,1<=K<=1000000，1<=d<=1000000000

Sample Input

8 7 4
1 2
2 3
3 4
5 6
6 7
7 8
8 5
2 3 1
1 4 1
5 5 8
1 8 10

Sample Output

TAK
NIE
TAK
NIE

题解：由于可以走重复的边，所以我们可以先走最短路过去，然后不断重复走同一条边。然而这样就对最短路长度的奇偶性有要求。所以我们将图分层，维护每个点到其他点的长度为奇数和偶数的最短路，用BFS即可。

然而本题卡空间所以要将询问离线。

还要特判从一个点走到自身，且这个点没有出边的情况。

#include <cstdio>

#include <cstring>

#include <iostream>

#include <queue>

using namespace std;

const int maxn=5010;

int n,m,k,cnt;

int to[maxn<<1],next[maxn<<1],head[maxn],dis[maxn][2],ans[1000010];

queue<int> qx,qy;

struct node

{

	int x,v,org;

	node() {}

	node(int a,int b,int c) {x=a,v=b,org=c;}

};

vector<node> p[maxn];

inline int rd()

{

	int ret=0,f=1;	char gc=getchar();

	while(gc<'0'||gc>'9')	{if(gc=='-')	f=-f;	gc=getchar();}

	while(gc>='0'&&gc<='9')	ret=ret*10+(gc^'0'),gc=getchar();

	return ret*f;

}

inline void add(int a,int b)

{

	to[cnt]=b,next[cnt]=head[a],head[a]=cnt++;

}

int main()

{

	memset(head,-1,sizeof(head));

	register int i,j,a,b,c;

	n=rd(),m=rd(),k=rd();

	for(i=1;i<=m;i++)	a=rd(),b=rd(),add(a,b),add(b,a);

	for(i=1;i<=k;i++)	a=rd(),b=rd(),c=rd(),p[a].push_back(node(b,c,i));

	for(i=1;i<=n;i++)	if(p[i].size())

	{

		memset(dis,-1,sizeof(dis));

		dis[i][0]=0,qx.push(i),qy.push(0);

		while(!qx.empty())

		{

			a=qx.front(),b=qy.front(),qx.pop(),qy.pop();

			for(j=head[a];j!=-1;j=next[j])	if(dis[to[j]][b^1]==-1)

				dis[to[j]][b^1]=dis[a][b]+1,qx.push(to[j]),qy.push(b^1);

		}

		for(j=0;j<(int)p[i].size();j++)	if(dis[p[i][j].x][p[i][j].v&1]!=-1&&dis[p[i][j].x][p[i][j].v&1]<=p[i][j].v)

		{

			if(head[i]!=-1)	ans[p[i][j].org]=1;

		}

	}

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

	{

		if(ans[i])	printf("TAK\n");

		else	printf("NIE\n");

	}

	return 0;

}