题意:数组的awesomeness是数组可以拆分的连续相同数的最小块数。有一个长度为 n 的数组 a,有 m 次询问 i,x,将数组第 i 个数改为 x,输出数组 a 所有子串的 awesomeness 的和。
思路:所有最小块数之间也就是 ai != ai+1 我们称为断点,一个子串的 awesomeness 是这个子串中断点的个数加一,那么我们求出一个断点有多少个子串包含它,就是这个断点的贡献,再把所有断点的贡献加在一起,再加上数组 a 子串的个数就是答案。对于一个断点左边有 i 个数,右边有 n - i 个数,那么这个断点的贡献为i * (n - i)。每次修改一个值,如果减少了断点,那么就减去这个断点的贡献,如果增加了断点就加上这个断点的贡献。
代码:
#include<bits/stdc++.h>
#define pb push_back
#define all(x) x.begin(), x.end()
using namespace std;
typedef long long ll;
typedef pair<int, int> PII;
const int N = 2e5 + 10, P = 1e9 + 7, mod = 998244353;
ll a[N];
void solve(){
ll n, m;
cin >> n >> m;
for(int i = 1; i <= n; i++) cin >> a[i];
a[n + 1] = 0;
ll ans = 0;
for(int i = 1; i <= n; i++)
ans += (a[i] != a[i + 1]) * i * (n - i);
while(m--){
ll i, x;
cin >> i >> x;
ans -= (a[i] != a[i + 1]) * i * (n - i);
ans -= (a[i - 1] != a[i]) * (i - 1) * (n - i + 1);
a[i] = x;
ans += (a[i] != a[i + 1]) * i * (n - i);
ans += (a[i - 1] != a[i]) * (i - 1) * (n - i + 1);
cout << ans + (n + 1) * n / 2 << endl;
}
}
int main(){
ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
int tt;
//cin >> tt;
//while(tt--) {
solve();
//}
return 0;
}