题目
注意数据范围是 N ≤ 5000 N\leq 5000 N≤5000!!
分析
考虑最plain的DP,
d
p
[
u
]
[
v
]
=
1
/
0
dp[u][v]=1/0
dp[u][v]=1/0表示
u
u
u到
v
v
v是否有一条回文路径,如果
d
p
[
u
]
[
v
]
=
1
dp[u][v]=1
dp[u][v]=1,那么
d
p
[
x
]
[
y
]
=
1
dp[x][y]=1
dp[x][y]=1(
x
x
x为
u
u
u的邻结点,
y
y
y为
v
v
v的邻结点,且
N
u
m
x
=
N
u
m
y
Num_x=Num_y
Numx=Numy)。用Bfs
的顺序完成这个DP即可,大约是这样:
queue<Edge> Q;
int Ans[MAXN+5][MAXN+5];
inline void Push(Edge e){
Q.push(e);
Ans[e.u][e.v]=Ans[e.v][e.u]=1;
}
for(int i=1;i<=M;i++)
Push(E[i]);
while(!Q.empty()){
int u=Q.front().u,v=Q.front().v;Q.pop();
for(int i=0;i<int(G[u].size());i++){
int x=G[u][i];
for(int j=0;j<int(G[v].size());j++){
int y=G[v][j];
if(!Ans[x][y]&&Num[x]==Num[y])
Push((Edge){x,y});
}
}
}
于是我们喜闻乐见地发现,对于每条边最坏情况下会往外扩展另外的所有边,于是时间复杂度就成了
O
(
M
2
)
O(M^2)
O(M2),TLE,据说有30分。
然后,既然边太多了,我们就尝试砍掉一些边,使新图和原图对题意等价。
(由于删掉一些边后,原本为
0
0
0的
d
p
[
u
]
[
v
]
dp[u][v]
dp[u][v]现在肯定也为
0
0
0,所以我们只需要让原本为
1
1
1的
d
p
[
u
]
[
v
]
dp[u][v]
dp[u][v]不会变成
0
0
0即可)
对于一条路径,它一定是由若干段相同权值的点构成的(下图中点上的数字代表权值):
如果这条路径是回文的,则
∣
A
∣
=
∣
E
∣
|A|=|E|
∣A∣=∣E∣,
∣
B
∣
=
∣
D
∣
|B|=|D|
∣B∣=∣D∣。
注意题面里加粗的字,一条边是可以反复使用的,也就是说,如果 ∣ A ∣ = ∣ E ∣ − 2 |A|=|E|-2 ∣A∣=∣E∣−2,我们完全可以把一条边 ( u , v ) ∈ A (u,v)\in A (u,v)∈A多走一遍,那 A A A就由“ ⋯ → u → v → ⋯ \cdots\to u\to v\to\cdots ⋯→u→v→⋯”变成了“ ⋯ → u → v → u → v → ⋯ \cdots\to u\to v\to u\to v \to\cdots ⋯→u→v→u→v→⋯”,这样一来 ∣ A ∣ |A| ∣A∣就等于 ∣ E ∣ |E| ∣E∣了。
于是得到一个结论:只要 ∣ A ∣ |A| ∣A∣和 ∣ E ∣ |E| ∣E∣、 ∣ C ∣ |C| ∣C∣和 ∣ D ∣ |D| ∣D∣……奇偶性相同,这条路径就可以视为回文的。
好了,图中的奇偶性问题,果断考虑二分图。
对于一个只由相同权值的点(例如, 0 0 0)组成的连通块,分类讨论:
- 若它是二分图,那原来在这个联通块里面就一定奇数或偶数条边的路径之一:
例如原来某条回文路径如图所示,左进右出,那这条路径在该二分图里面,不管怎么走,一定都是奇数长度。也就是说,只要这个二分图是联通的,原路径2就可以转化成奇偶性相同的另一条路径。那么我们对该二分图做一个生成树,是不会影响 d p dp dp中的值的,边数大大减少。 - 如果不是二分图,原来在这个联通块里面就一定是奇数、偶数条边的路径都可以,所以做完生成树后随便找一个点,加个自环,就能保证奇数偶数长度的路径一定都有了,边数同样大大减少。
现在其实还有一类边:第一张图中的橘色边,这样的边怎么连是没有影响的(我们考虑的是点,考虑按同点权的点缩点后,这样的边一定是单个出现的,所以没有影响),所以直接生成树。
代码
#include<queue>
#include<cstdio>
#include<algorithm>
using namespace std;
int read(){
int x=0;char c=getchar();
while(c<'0'||c>'9')c=getchar();
while(c>='0'&&c<='9')x=x*10+(c^48),c=getchar();
return x;
}
#define MAXN 5000
#define MAXM 500000
#define MAXT 100000
int N,M,T;
int Num[MAXN+5];
char str[MAXN+5];
int Fa[MAXN+5];
int Find(int u){
return Fa[u]==u?u:Fa[u]=Find(Fa[u]);
}
struct Edge{
int u,v;
}E[MAXM+5];
vector<int> G[MAXN+5],H[MAXN+5];
queue<Edge> Q;
int Ans[MAXN+5][MAXN+5];
inline void Push(Edge e){
Q.push(e);
Ans[e.u][e.v]=Ans[e.v][e.u]=1;
}
int Col[MAXN+5];
bool Check(int u,int c){
Col[u]=c;
bool ret=1;
for(int i=0;i<int(H[u].size());i++){
int v=H[u][i];
if(Num[v]==Num[u]){
if(!Col[v]){
G[u].push_back(v);
G[v].push_back(u);
Push((Edge){u,v});
ret&=Check(v,c^1);
}
else if(Col[v]==c)
ret=0;
}
}
return ret;
}
int main(){
N=read(),M=read(),T=read();
scanf("%s",str+1);
for(int i=1;i<=N;i++)
Num[i]=str[i]-'0';
for(int i=1;i<=M;i++){
int u=read(),v=read();
E[i].u=u,E[i].v=v;
H[u].push_back(v);
H[v].push_back(u);
}
for(int i=1;i<=N;i++)
Fa[i]=i;
for(int i=1;i<=N;i++)
if(!Col[i]){
if(!Check(i,2))
G[i].push_back(i);
}
for(int i=1;i<=M;i++){
int u=E[i].u,v=E[i].v;
if(Num[u]!=Num[v]&&Find(u)!=Find(v)){
Fa[Find(u)]=Find(v);
G[u].push_back(v);
G[v].push_back(u);
}
}
for(int i=1;i<=N;i++)
Push((Edge){i,i});
while(!Q.empty()){
int u=Q.front().u,v=Q.front().v;Q.pop();
for(int i=0;i<int(G[u].size());i++){
int x=G[u][i];
for(int j=0;j<int(G[v].size());j++){
int y=G[v][j];
if(!Ans[x][y]&&Num[x]==Num[y])
Push((Edge){x,y});
}
}
}
while(T--)
puts(Ans[read()][read()]?"YES":"NO");
}