题目大意:给定一张无向图,每条边边权都是1,多次询问是否存在某个点到达另一个点的长度为d的路径
首先如果s和t是同一点且这个点没有出边 那么s到t只存在长度为0的路径
否则:
如果s到t有长度为d的路径 那么就一定有长度为d+2的路径
因此只要BFS求出s开始到每个点的奇数长度的最短路和偶数长度的最短路就行了
为了防止爆内存可以预先将询问离线化
时间复杂度O(n(n+m))
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define M 5050
using namespace std;
struct Query{
int s,t,d;
Query() {}
Query(int _,int __,int ___):
s(_),t(__),d(___) {}
}queries[1001001],*q[1001001];
struct abcd{
int to,next;
}table[M<<1];
int head[M],tot;
int n,m,k;
int f[M][2];
bool ans[1001001];
void Add(int x,int y)
{
table[++tot].to=y;
table[tot].next=head[x];
head[x]=tot;
}
bool Compare(Query *x,Query *y)
{
return x->s < y->s;
}
void BFS(int x)
{
static pair<int,bool> q[M<<1];
int i,r=0,h=0;
memset(f,0x3f,sizeof f);
f[x][0]=0;
q[++r]=make_pair(x,0);
while(r!=h)
{
pair<int,bool> temp=q[++h];
int x=temp.first;
bool flag=temp.second;
for(i=head[x];i;i=table[i].next)
if(f[table[i].to][!flag]==0x3f3f3f3f)
f[table[i].to][!flag]=f[x][flag]+1,q[++r]=make_pair(table[i].to,!flag);
}
}
int main()
{
int i,j,x,y,d;
cin>>n>>m>>k;
for(i=1;i<=m;i++)
{
scanf("%d%d",&x,&y);
Add(x,y);Add(y,x);
}
for(i=1;i<=k;i++)
{
scanf("%d%d%d",&x,&y,&d);
q[i]=new (&queries[i])Query(x,y,d);
}
sort(q+1,q+k+1,Compare);
for(j=1,i=1;i<=n;i++)
{
BFS(i);
for(;j<=k&&q[j]->s==i;j++)
{
x=q[j]->s;
y=q[j]->t;
d=q[j]->d;
if( x==y && !head[x] )
ans[q[j]-queries]=!d;
else
ans[q[j]-queries]=f[y][d&1]<=d;
}
}
for(i=1;i<=k;i++)
puts(ans[i]?"TAK":"NIE");
return 0;
}