Solution -「ARC 126E」Infinite Operations

\(\mathcal{Description}\)

  Link.

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

• 选择 \(a_i<a_j\)，以及一个 \(x\in\mathbb R_+\)，使得 \(a_i+x\le a_j-x\)；
• 令 \(a_i\leftarrow a_i+x,a_j\leftarrow a_j-x\)，本次操作的得分为 \(x\)。

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

\(\mathcal{Solution}\)

  定义序列的势能 \(\Phi=\sum_{i<j}|a_i-a_j|\)，设进行一次操作后得到新势能 \(\Phi'\)，显然有 \(\Phi'-\Phi\le-2x\)，其中 \(x\) 即操作时所选取的正数。同时，取等条件容易看出为 \(\not\exist a_k,a_k\in(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\n", mul( ans, INV2 ) );
	}
	return 0;
}