欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

CodeForces 1320D - Reachable Strings

程序员文章站 2022-03-26 11:49:45
"洛谷题目页面传送门" & "CodeForces题目页面传送门" 有一个01串$a,|a|=n$。$q$次询问,每次给出$l1,l2,len$,问$a_{l1\sim l1+len 1},a_{l2\sim l2+len 1}$这$2$个01串是否能通过若干次操作变得相等,其中一次操作指的是将任意 ......

洛谷题目页面传送门 & codeforces题目页面传送门

有一个01串\(a,|a|=n\)\(q\)次询问,每次给出\(l1,l2,len\),问\(a_{l1\sim l1+len-1},a_{l2\sim l2+len-1}\)\(2\)个01串是否能通过若干次操作变得相等,其中一次操作指的是将任意一个01串的任意一个等于\(\texttt{110}\)的子串变成\(\texttt{011}\)或将\(\texttt{011}\)变成\(\texttt{110}\)

\(n,q\in\left[1,2\times10^5\right]\)

考虑分析能通过若干次操作变得相等的充要条件。

不难发现,每次操作都不可能改变01串中\(\texttt0\)\(\texttt1\)分别的数量,所以\(2\)个01串中\(\texttt0\)\(\texttt1\)的数量分别相等是能通过若干次操作变得相等的必要条件,但充分性还远远不够。

又发现,\(\texttt{110}\leftrightarrow\texttt{011}\)这个操作比较有意思,它相当于将这个唯一的\(\texttt{0}\)向左/右推了\(2\)位。由于这是个01串,非\(\texttt0\)\(\texttt1\),于是我们将所有\(\texttt0\)的位置揪出来,其他位置自然是\(\texttt1\)

不难发现,01串中所有\(\texttt0\)对的相对位置都不会改变。证明:若\(2\)\(\texttt0\)想要交换位置,那么交换前最后一刻的状态一定是距离差\(\leq2\),此时左边的\(\texttt0\)不能通过\(\texttt{011}\to\texttt{110}\)往右再移\(2\)格,因为它右边\(1\sim2\)格是右边的\(\texttt0\),以它开头的长度为\(3\)的子串一定不为\(\texttt{011}\)。类似地,右边的\(\texttt0\)也不能往左移\(2\)格。得证。

于是考虑将\(2\)个01串中的所有\(\texttt0\)按出现次序一一对应。显然,第\(1\)个01串中的某个\(\texttt0\)能够与第\(2\)个01串中的某个\(\texttt0\)到同一位置上当且仅当它们在所在串中的位置之差为偶数,即奇偶性相等。所以我们猜测:\(2\)个01串中的所有\(\texttt0\)按出现次序一一对应后,每对\(\texttt0\)在所在串中的位置奇偶性相等,是这\(2\)个01串能通过若干次操作变得相等的充要条件。证明:

  1. 充分性证明:数学归纳法。
    1. \(2\)个串都不存在\(\texttt0\)时,充分性显然;
    2. 假设当\(2\)个串都存在\(x-1(x\geq1)\)\(\texttt0\)时,满足充分性。此时若要将\(2\)个存在\(x\)\(\texttt0\)的01串的所有\(\texttt0\)对一一对齐,可以将左数第\(1\)\(\texttt0\)中较右的那个\(\texttt0\)通过若干次操作与较左的对齐,问题转化为将剩下\(x-1\)\(\texttt0\)一一对齐,根据假设,存在方案。所以若当\(2\)个串都存在\(x-1\)\(\texttt0\)时,满足充分性,那么当\(2\)个串都存在\(x\)\(\texttt0\)时,满足充分性。
    综上,充分性得证;
  2. 必要性显然。

综上,得证。

接下来问题就变成了比较裸的序列哈希:每次给定\(2\)个子串,问这\(2\)个子串中\(\texttt0\)的位置奇偶性组成的序列是否相等。注意:这里的位置奇偶性指的是相对于\(l1/l2\)的位置奇偶性,而不是相对于\(1\),所以要记录\(2\)个每项对应相反的\(\texttt0\)的位置奇偶性序列。子串中包含的\(\texttt0\)的位置奇偶性序列的子序列可以二分查找找到。

下面是ac代码:

#include<bits/stdc++.h>
using namespace std;
#define mp make_pair
#define pb push_back
const int n=200000;
int n/*01串长度*/,qu/*询问次数*/;
char a[n+5];//01串 
vector<int> pos;//'0'的位置序列 
const int hbase1=131,hmod1=998244353,hbase2=13331,hmod2=1000000007;//哈希参数 
int hsh1[n+1],rhsh1[n+1],hpw1[n+1],hsh2[n+1],rhsh2[n+1],hpw2[n+1];//hsh:相对于1的'0'的位置奇偶性序列的前缀哈希,rhsh:相对于2的…… 
void hsh_init(){//哈希预处理 
    hpw1[0]=hpw2[0]=1;
    for(int i=1;i<=pos.size();i++)//为了前缀运算方便,哈希数组们1-indexed 
        hsh1[i]=(1ll*hsh1[i-1]*hbase1+pos[i-1]%2+1)%hmod1,
        rhsh1[i]=(1ll*rhsh1[i-1]*hbase1+!(pos[i-1]%2)+1)%hmod1,
        hpw1[i]=1ll*hpw1[i-1]*hbase1%hmod1,
        hsh2[i]=(1ll*hsh2[i-1]*hbase2+pos[i-1]%2+1)%hmod2,
        rhsh2[i]=(1ll*rhsh2[i-1]*hbase2+!(pos[i-1]%2)+1)%hmod2,
        hpw2[i]=1ll*hpw2[i-1]*hbase2%hmod2;
}
pair<int,int> hsh(int l,int r){//pos[l-1~r-1]相对于1的奇偶性序列的哈希值 
    return mp(((hsh1[r]-1ll*hsh1[l-1]*hpw1[r-l+1]%hmod1)+hmod1)%hmod1,
              ((hsh2[r]-1ll*hsh2[l-1]*hpw2[r-l+1]%hmod2)+hmod2)%hmod2);
}
pair<int,int> rhsh(int l,int r){//pos[l-1~r-1]相对于2的奇偶性序列的哈希值
    return mp(((rhsh1[r]-1ll*rhsh1[l-1]*hpw1[r-l+1]%hmod1)+hmod1)%hmod1,
              ((rhsh2[r]-1ll*rhsh2[l-1]*hpw2[r-l+1]%hmod2)+hmod2)%hmod2);
}
int main(){
    cin>>n>>a+1>>qu;
    for(int i=1;i<=n;i++)if(a[i]=='0')pos.pb(i);//预处理pos 
    hsh_init();//预处理哈希 
    while(qu--){
        int l1,l2,len;
        scanf("%d%d%d",&l1,&l2,&len);
        int l1_0=lower_bound(pos.begin(),pos.end(),l1)-pos.begin()+1,//a[l1~l1+len-1]包含的'0'的位置奇偶性序列的左端点 
            r1_0=lower_bound(pos.begin(),pos.end(),l1+len)-pos.begin(),//a[l1~l1+len-1]包含的'0'的位置奇偶性序列的右端点
            l2_0=lower_bound(pos.begin(),pos.end(),l2)-pos.begin()+1,//a[l2~l2+len-1]包含的'0'的位置奇偶性序列的左端点
            r2_0=lower_bound(pos.begin(),pos.end(),l2+len)-pos.begin();//a[l2~l2+len-1]包含的'0'的位置奇偶性序列的右端点
        pair<int,int> hsh1=l1%2?hsh(l1_0,r1_0):rhsh(l1_0,r1_0),hsh2=l2%2?hsh(l2_0,r2_0):rhsh(l2_0,r2_0);//计算哈希值 
        puts(hsh1==hsh2?"yes":"no");//判断相等 
    }
    return 0;
}