CF1270F Awesome Substrings 题解

题目来源

• CF1270F

• 洛谷

Description

给定一个长度为 $n$ 的 $0/1$ 串 $s$，求有多少个区间 $[l,r]$ 满足 $r-l+1$ 是 $s_{l\cdots r}$ 中 $1$ 的个数的倍数。

• $n\le 2\times 10^5$。

Solution

首先看到倍数问题，可以考虑 $\text{polylog}$ 做法和根号分治的做法。

但本题似乎只能使用根号分治的做法，于是我们可以枚举倍数 $k$。

当 $k\le \sqrt{n}$ 时，我们观察到如果设 $cn{t}_0,cn{t}_1$ 分别为区间内 $0$ 和 $1$ 的个数，则应满足 $kcn{t}_1=cn{t}_0+cn{t}_1$。

于是我们可以把 $s$ 中所有的 $1$ 全部赋值为 $k-1$，$0$ 则赋值为 $-1$，则一个区间权值和为 $0$ 和其满足 $kcn{t}_1=cn{t}_0+cn{t}_1$ 互为充要条件。

于是，我们就可以进行前缀和，并开一个桶统计即可（由于前缀和可能为负，需要加个偏移量 $n$）。

而当 $k>\sqrt{n}$ 时，$1$ 的可能的个数最多为 $\frac{n}{k}\le \sqrt{n}$。

因此，我们可以枚举区间的左端点 $l$ 以及 $1$ 的个数 $i$，并计算出满足条件的区间右端点所在的范围 $[r_l,r_r]$。

如何计算？只需要预处理出第 $i$ 个 $1$ 所在的位置 $po{s}_i$，便可 $O(1)$ 计算。

则满足条件的区间长度范围为 $[r_l-l+1,r_r-l+1]$，可能的倍数范围则为 $[\lceil \frac{r_l-l+1}{i}\rceil ,\lfloor \frac{r_r-l+1}{i}\rfloor ]$。

其对答案的贡献即为 ⌊ r r − l + 1 i ⌋ − max ⁡ { n , ⌊ r l − l i ⌋ } \lfloor\dfrac{r_r-l+1}i\rfloor-\max\{\sqrt n,\lfloor\dfrac{r_l-l}i\rfloor\} ⌊irr​−l+1​⌋−max{n ​,⌊irl​−l​⌋}，后面部分取 $\max$ 的原因是应把 $k\le \sqrt{n}$ 的部分去除。

最后输出答案即可，整体的复杂度为 $O(n\sqrt{n})$。

Code

#include <bits/stdc++.h>
using namespace std;
int n,B,sum[200005],pos[200005],cnt[200005*505];
long long ans;
char s[200005];
int main(){
    scanf("%s",s+1);
    n=strlen(s+1),B=sqrt(n); // 根号分治
    for (int k=1;k<=B;k++){ // k<=根号n
        cnt[n]=1;
        for (int i=1;i<=n;i++) sum[i]=sum[i-1]+(s[i]=='1'?k-1:-1),ans+=cnt[sum[i]+n],cnt[sum[i]+n]++; // 1赋值为k-1，0赋值为-1，则符合条件的区间权值和为0
        for (int i=1;i<=n;i++) cnt[sum[i]+n]--,sum[i]=0;
    }
    for (int i=1;i<=n;i++)
        if (s[i]=='1') sum[i]=sum[i-1]+1,pos[sum[i]]=i;
        else sum[i]=sum[i-1];
    pos[sum[n]+1]=n+1;
    for (int l=1;l<=n;l++) // 左端点
        for (int i=1;i<=n/B;i++){ // 1的个数
            int rl=pos[i+sum[l-1]],rr=pos[i+sum[l-1]+1]-1;
            if (rl>0&&rr>0) ans+=max((rr-l+1)/i-max(B,(rl-l)/i),0);
        }
    printf("%lld\n",ans);
    return 0;
}