BZOJ 4373 算术天才⑨与等差数列 [线段树]

题意：给出n个数，有两个操作：1、可以对数组其中的某一个数进行修改 2、询问数组中区间为【L，R】的数从小到大排列能否组成等差数列

题解：是否为等差数列有两个条件：1、数的总和要等于 （首项+末项）*项数/2   2、数的平方和要等于 n*mi*mi+n*（n-1）*（2*n-1）*d*d/6+n*（n-1）*d*mi；这两个都可以通过线段树来维护。（注意平方和要爆longlong 要取mod）

AC代码：

#include<stdio.h>
#include<set>
#define mod 1000000007
#define N 300005
using namespace std;
typedef long long ll;
ll a[N];
ll sum[N*4],trmin[N*4],pfsum[N*4];
ll inv;
void build(ll L,ll R,ll root)
{
	if(L==R)
	{
		sum[root]=a[L];
		trmin[root]=a[L];
		pfsum[root]=a[L]*a[L]%mod;
		return ;
	}
	ll mid=L+R>>1;
	build(L,mid,root<<1);
	build(mid+1,R,root<<1|1);
	sum[root]=sum[root<<1]+sum[root<<1|1];
	trmin[root]=min(trmin[root<<1],trmin[root<<1|1]);
	pfsum[root]=(pfsum[root<<1]+pfsum[roo