> 文章列表 > Codeforces Round 816 (Div. 2)(C. Monoblock)

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

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

传送门

 

题意:

定义一个数组,定义它 相等连续段的个数
如1,2,2,1,相等连续段有[1],[2,2],[1],共3个
如2,2,2,1,相等连续段有[2,2,2],[1],共2个

给定一个数组a,定义g(l,r)为,子数组中a[l],a[l+1],…,a[r],相等连续段的个数。给定m次操作,每次修改数组第i个位置的值为x。求每次操作后的数组的g函数的值。

f函数定义为:\\sum_{l=1}^{n}\\sum_{r=l}^{n} g(l,r)

 

思路
先来看看全部相等的元素的数组,它的h函数的值。
那么

            f=  \\sum_{l=1}^{n}\\sum_{r=l}^{n} g(l,r) = \\sum_{l=1}^{n}\\sum_{r=l}^{n}f=\\sum_{l=1}^{n}\\sum_{r=l}^{n}g(l,r)=\\sum_{l=1}^{n}\\sum_{r=l}^{n}=n*(n+1)/2

 

对于一个元素全部相等的区间,它的g函数值为1,
如果修改一个元素,且该元素与其相邻元素不等时,那么该区间的g函数值会加1。

1. 计算初始的h函数值。
我们把数组初始看成n个a[1]。
此时h函数为n*(n+1)/2。

接着把第二个数修改为a[2],此时,如果a[2]!=a[1],那么所有包含a[2]和a[1]的区间,g函数都会加1。

这里注意防爆int,转换为long long做计算

AC代码:

#include<bits/stdc++.h>
//#define int long long
using namespace std;
using i64=long long;
const int N=1e6+10;
i64 a[N];
void solve(){i64 n,m;cin>>n>>m;i64 sum=n*(n-1)/2+n;for(int i=1;i<=n;i++) cin>>a[i]; for(i64 i=1;i<=n-1;i++){
//		cin>>a[i];if(a[i]!=a[i+1]){sum+=(i)*(n-i); }}while(m--){i64 i,x;cin>>i>>x;if(a[i]!=a[i-1]) sum-=(i-1)*(n-(i-1));if(a[i]!=a[i+1]) sum-=i*(n-i);a[i]=x;if(a[i]!=a[i-1]) sum+=(i-1)*(n-(i-1));if(a[i]!=a[i+1]) sum+=i*(n-i);cout<<sum<<'\\n';}}
signed main(){ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);int t;
//	cin>>t;t=1;while(t--){solve();}
}