题面
分析
预处理
先考虑在没有修改的情况下的初值,下面以 {1, 2, 2, 4, 5} 为例:
我们可以先处理从 1 开始的贡献,这点很显然。
我们再观察从下图中 第 i 行 到 第 i + 1 行 的和贡献变化 (下图中第 i 行第 j 列表示从 i 到 j 的贡献),我么可以发现,如果 a [ i ] = = a [ i + 1 ] a[i] == a[i +1] a[i]==a[i+1] 那么从这两行的差值就为 1,而如果 a [ i ] ≠ a [ i + 1 ] a[i] \neq a[i + 1] a[i]=a[i+1],则这两行的差值就为 i。
for(int i = 1; i <= n; i++) {
a[i] = inpt();
if(a[i] != a[i - 1])
nw++;
ans += nw, tmp += nw;
}
for(int i = 2; i <= n; i++) {
if(a[i<