【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;
}