洛谷P2757 [国家集训队]等差子序列

https://www.luogu.com.cn/problem/P2757

题意简化就是找到a[i] a[j] a[k]成一个len=3的等差数列，且a[j]-a[i]=a[k]-a[j]

那么枚举a[j],为方便从左到右枚举，设数字x在j左边为1，在右边为0

那么可以用树状数组维护一个hash值，如果a[j]+d和a[j]-d在不同边也就是一个是0一个是1，那么就存在一组等差序列了

也就是这个hash值如果不相等，说明某一对a[j]+d和a[j]-d分别在j的左边和右边，那么就有解

这个hash值要维护一个正着的一个倒着的，因为一边是+d一边是-d，我们要让a[j]-1和a[j]+1一一对应起来，就是将第一个树状数组b1的a[j]-1与b2的n-a[j]对齐，然后短的那边补成场的也就是乘以hash的base的那么多次幂，长的那边截取短的长度，看这两个值一不一样

 

#include<bits/stdc++.h>
using namespace std;
typedef unsigned long long ull;

const int maxl=1e5+10;
const int bas=1423333;

int n,ans;
int a[maxl],pre[maxl],suf[maxl];
ull mi[maxl];
struct bit
{
	ull b[maxl];
	inline void init()
	{
		for(int i=1;i<=n;i++)
			b[i]=0;
	}
	inline void add(int i,ull x)
	{
		while(i<=n)
			b[i]+=x,i+=i&-i;
	}
	inline ull sum(int i)
	{
		ull ret=0;
		while(i)
			ret+=b[i],i-=i&-i;
		return ret;
	}