bzoj 2124 等差子序列 字符串哈希+树状数组

题面

题目传送门

解法

思路极其精妙……

• 显然，题目就是问是否能找到一个长度为$3$的等差数列，因为长度大于$3$的等差数列中也一定存在长度为$3$的等差数列
• 那么，问题就变成枚举一个$a[j]$，是否存在$a[i]-a[j]=a[j]-a[k]$
• 因为这是一个排列，不存在相同的数，那么直接开一个桶就可以用最坏复杂度为$O(Tn^2)$的算法水过去
• 当然，$O(n^2)$的复杂度来解决这个问题似乎太简单了，如果$n$到$10^5$级别那么这样就做不了了
• 我们令$b[i]$数组表示枚举到$a[j]$的时候，$i$这个数是否出现过
• 显然，如果$b[a[j]-x]≠b[a[j]+x]$，那么这三个数一定可以构成一个等差序列
• 所以，我们的问题就变成这样一个问题：每一次把一个位置$x$修改成$1$，然后询问以$x$为中心的最长子串是否回文
• 用树状数组来维护哈希值即可
• 时间复杂度：$O(Tn\log n)$

代码

#include <bits/stdc++.h>
#define Mod 1000000009
#define N 10010
using namespace std;
template <typename node> void chkmax(node &x, node y) {x = max(x, y);}
template <typename node> void chkmin(node &x, node y) {x = min(x, y);}
template <typename node> void read(node &x) {
    x = 0; int f = 1; char c = getchar();
    while (!isdigit(c)) {if (c == '-') f = -1; c = getchar();}
    while (isdigit(c)) x = x * 10 + c - '0', c = getchar(); x *= f;
}
int n, a[N], s[N];
struct BIT {
    int f[N];
    int lowbit(int x) {return x & -x;}
    void Clear() {memset(f, 0, sizeof(f));}
    void modify(int x) {
        for (int i = x; i <= n; i += lowbit(i))
            f[i] = (f[i] + s[i - x]) % Mod;
    }
    int query(int x) {
        int ret = 0;
        for (int i = x; i; i -= lowbit(i))
            ret = (ret + 1ll * f[i] * s[x - i] % Mod) % Mod;
        return ret;
    }
    int calc(int l, int r) {
        if (l > r) return 0;
        int x = query(r), y = query(l - 1);
        return (x - 1ll * s[r - l + 1] * y % Mod + Mod) % Mod;
    }
} t1, t2;
int main() {
    int T; read(T);
    while (T--) {
        read(n); s[0] = 1;
        for (int i = 1; i <= n; i++) read(a[i]), s[i] = 1ll * s[i - 1] * 233 % Mod;
        t1.Clear(), t2.Clear(); bool flag = false;
        for (int i = 1; i <= n; i++) {
            int len = min(a[i] - 1, n - a[i]);
            int x = t1.calc(a[i] - len, a[i] - 1), y = t2.calc(n - a[i] - len + 1, n - a[i]);
            if (x != y) flag = true;
            t1.modify(a[i]);
            t2.modify(n - a[i] + 1);
        }
        puts((flag) ? "Y" : "N");
    }
    return 0;
}