Codeforces Round #816 (Div. 2) C. Monoblock

 input:

5 5
1 2 3 4 5
3 2
4 2
3 1
2 1
2 2
 

output:

29
23
35
25
35
题目大意：给出一个长度为n的数组，有m个询问，每次询问给出2个数字，p,x，表示让a[p]=x，然后定义g(l,r)表示a[l]~a[r]这段数组可以分成多少段连续相等的段，例如999111222就可以分成3组连续相等的段分别是999,111,222，对于每一个询问给出所有区间的g(l,r)之和。

解题思路：首先我们想象一开始整个数组都是a[1]，那么每一段可以分成1段连续相等的段，一共有n*(n+1)/2个区间，然后我们开始逐个对a数组进行改变，对于初始数组，如果当前第i个数字和第i-1个数字不相等，那么包含这两个数的所有区间的g值都要加一，也就是答案要加上(i-1)*(n-i+1)，然后对于每次询问，修改某个位置p可能会导致a[p-1]与a[p]，a[p]与a[p+1]的相等关系发生变化，拿a[p-1]和a[p]来说，修改了a[p]之后，可能会使两者由原来相等的情况变成不相等的情况，也有可能会使两者由原来不相等的情况变成相等的情况，前一种情况下，所有包含这两个元素的区间值都要加一，后一种情况下，所有包含这两个元素的区间值都要减一，所以对于每个询问我们都可以用O(1)的时间来得到答案。

上代码：

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
using namespace std; 
const int N=1e5+10;
int a[N];
long long n,m,ans;
void check(int l,int r)
{
	ans++;
	for(int i=l+1;i<=r;i++)
	if(a[i]!=a[i-1])
	ans++;
}
int main()
{
	cin>>n>>m;
	ans=n*(n+1)/2;
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
		if(a[i]!=a[i-1])
		ans+=1ll*(i-1)*(n-i+1);
	}
	for(int i=1;i<=m;i++)
	{
		int pos,x;
		cin>>pos>>x;
		if(x==a[pos])
		{
			cout<<ans<<endl;
			continue;
		}
		if(pos>1&&a[pos]==a[pos-1]&&x!=a[pos-1])
		ans+=1ll*(pos-1)*(n-pos+1);
		if(pos<n&&a[pos]==a[pos+1]&&x!=a[pos+1])
		ans+=1ll*pos*(n-pos);
		if(pos>1&&a[pos]!=a[pos-1]&&x==a[pos-1])
		ans-=1ll*(pos-1)*(n-pos+1);
		if(pos<n&&a[pos]!=a[pos+1]&&x==a[pos+1])
		ans-=1ll*pos*(n-pos);
		cout<<ans<<endl;
		a[pos]=x;
	}
	return 0;
}