Solution -「ARC 126E」Infinite Operations

(mathcal{Description})

  Link.

  给定序列 ({a_n})，定义一次操作为：

• 选择 (a_i<a_j)，以及一个 (xinmathbb R_+)，使得 (a_i+xle a_j-x)；
• 令 (a_ileftarrow a_i+x,a_jleftarrow a_j-x)，本次操作的得分为 (x)。

  定义序列的得分为进行任意次操作能得到的最大得分和，现给定 (m) 次形如 (a_xleftarrow y) 的修改操作，在每次修改操作后求出当前序列的得分。

(mathcal{Solution})

  定义序列的势能 (Phi=sum_{i<j}|a_i-a_j|)，设进行一次操作后得到新势能 (Phi')，显然有 (Phi'-Phile-2x)，其中 (x) 即操作时所选取的正数。同时，取等条件容易看出为 ( otexist a_k,a_kin(a_i,a_j))，由此，又能得到只要 (Phi>0)，我们必然能以势能减少 (2x) 的代价增加 (x) 分。所以答案就是 (frac{Phi}{2})。

  选用平衡树来动态维护 (Phi) 即可支持修改。复杂度 (mathcal O((n+m)log n))。

(mathcal{Code})

/*~Rainybunny~*/

#include <bits/stdc++.h>

#define rep( i, l, r ) for ( int i = l, rep##i = r; i <= rep##i; ++i )
#define per( i, r, l ) for ( int i = r, per##i = l; i >= per##i; --i )

typedef long long LL;

const int MAXN = 6e5, MOD = 998244353, INV2 = 499122177;
int n, m, root, a[MAXN + 5];

inline int mul( const int u, const int v ) { return 1ll * u * v % MOD; }
inline int sub( int u, const int v ) { return ( u -= v ) < 0 ? u + MOD : u; }
inline int add( int u, const int v ) { return ( u += v ) < MOD ? u : u - MOD; }

struct Treap {
	int node, root, ch[MAXN + 5][2], aux[MAXN + 5], val[MAXN + 5];
	int siz[MAXN + 5], sum[MAXN + 5];
	Treap() { srand( 20120712 ); }
	
	inline int newnd( const int v ) {
		++node, aux[node] = rand(), sum[node] = val[node] = v, siz[node] = 1;
		return node;
	}
	
	inline void pushup( const int u ) {
		siz[u] = siz[ch[u][0]] + siz[ch[u][1]] + 1;
		sum[u] = add( add( sum[ch[u][0]], sum[ch[u][1]] ), val[u] );
	}
	
	inline int merge( const int u, const int v ) {
		if ( !u || !v ) return u | v;
		if ( aux[u] < aux[v] ) {
			return ch[u][1] = merge( ch[u][1], v ), pushup( u ), u;
		} else {
			return ch[v][0] = merge( u, ch[v][0] ), pushup( v ), v;
		}
	}
	
	inline void vsplit( const int u, const int k, int& x, int& y ) {
		if ( !u ) return void( x = y = u );
		if ( val[u] <= k ) vsplit( ch[u][1], k, ch[x = u][1], y ), pushup( x );
		else vsplit( ch[u][0], k, x, ch[y = u][0] ), pushup( y );
	}
	
	inline int erase( const int v ) {
		int x, y, z; vsplit( root, v, x, z );
		int ret = add( sub( mul( sub( siz[x], siz[z] ), v ), sum[x] ), sum[z]);
		vsplit( x, v - 1, x, y ), y = merge( ch[y][0], ch[y][1] );
		return root = merge( x, merge( y, z ) ), ret;
	}
	
	inline int insert( const int v ) {
		int x, y = newnd( v ), z; vsplit( root, v, x, z );
		int ret = add( sub( mul( sub( siz[x], siz[z] ), v ), sum[x] ), sum[z]);
		return root = merge( x, merge( y, z ) ), ret;
	}
} trp;

int main() {
	scanf( "%d %d", &n, &m );
	int ans = 0;
	rep ( i, 1, n ) scanf( "%d", &a[i] ), ans = add( ans, trp.insert( a[i] ) );
	for ( int x, v; m--; ) {
		scanf( "%d %d", &x, &v );
		ans = sub( ans, trp.erase( a[x] ) );
		ans = add( ans, trp.insert( a[x] = v ) );
		printf( "%d
", mul( ans, INV2 ) );
	}
	return 0;
}