给定数组a,定义g(l,r)为区间[l,r]相等连续段的个数。有q个询问,每次将a[i]修改为x。对于每个询问求:
思路:涉及到修改操作,提示我们可以从改变量入手。假设a数组全部元素都为a[1],现将a[2]改为a[2]。那么所有包含a[1],a[2]的区间都会加上一。
对于每个询问,判断改变后和相邻元素的关系,修改格式即可。
Code:
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
#include<cmath>
#include<vector>
#include<queue>
#include<stack>
#include<map>
#include<set>
#include<unordered_map>
using namespace std;
#define PII pair<int,int>
#define endl "\n"
typedef long long LL;
typedef unsigned long long ULL;
const int N=1e5+10;
int a[N];
void solve() {
int n,m;
cin>>n>>m;
LL res=1LL*n*(n+1)/2;
for(int i=1;i<=n;i++) {
cin>>a[i];
}
for(int i=1;i<n;i++) { //增加量
if(a[i]!=a[i+1]) {
res+=1LL*i*(n-i);
}
}
while(m--) {
int i,x;
cin>>i>>x;
res-=1LL*(a[i]!=a[i-1])*(n-i+1)*(i-1);
res-=1LL*(a[i]!=a[i+1])*(n-i)*i;
a[i]=x;
res+=1LL*(a[i]!=a[i-1])*(n-i+1)*(i-1);
res+=1LL*(a[i]!=a[i+1])*(n-i)*i;
cout<<res<<endl;
}
}
int main() {
ios::sync_with_stdio(false);
cin.tie(nullptr);
cout.tie(nullptr);
solve();
return 0;
}