Good problem

Average problem

Bad problem

Did not solve

#include<bits/stdc++.h>
using namespace std;
string s;
int main(){
	ios::sync_with_stdio(false),cin.tie(0);
	int T,n,i,ans;
	for(cin>>T;T>0;T--)
	{
		cin>>s;n=s.size();
		ans=0;
		for(i=0;i<n;i++)ans+=s[i]-'0';
		cout<<ans<<'\n';
	}
	return 0;
}

for _ in range(int(input())):
    print(input().count('1'))

Good problem

Average problem

Bad problem

Did not solve

#include<bits/stdc++.h>
using namespace std;
int main(){
	ios::sync_with_stdio(false),cin.tie(0);
	int T,n,i,t,flag;
	for(cin>>T;T>0;T--)
	{
		cin>>n;
		flag=0;
		for(i=0;i<n;i++)
		{
			cin>>t;
			if(t<=i*2||t<=(n-i-1)*2)flag=1;
		}
		if(flag)cout<<"NO\n";
		else cout<<"YES\n";
	}
	return 0;
}

for _ in range(int(input())):
    n=int(input())
    a=list(map(int,input().split()))
    for i in range(n):
        if a[i]<=max(i,n-i-1)*2:
            print('NO')
            break
    else:
        print('YES')

Good problem

Average problem

Bad problem

Did not solve

#include <bits/stdc++.h>
using namespace std;
 
long long a[1010];
 
int main(){
	int t;
	cin >> t;
	while (t--){
		int n;
		cin >> n;
		for (int i = 1; i <= n; i++) cin >> a[i];
		int now = n;
		long long ans = -1e18;
		for (int i = 1; i <= n; i++){
			long long sum = 0;
			for (int i = 1; i <= now; i++) sum = (sum + a[i]) ;
			if (i == 1) ans = max(ans, sum);
			else ans = max(ans, max(sum, ( - sum) ));
			for (int i = 1; i < now; i++) a[i] = (a[i + 1] - a[i]) ;
			now--;
		}
		cout << ans << endl; 
	}
	return 0;
}

from math import *
for _ in range(int(input())):
	n=int(input())
	a=list(map(int,input().split()))
	ans=sum(a)
	while n>1:
		n-=1
		a=[a[i+1]-a[i] for i in range(n)]
		ans=max(ans,abs(sum(a)))
	print(ans)

Good problem

Average problem

Bad problem

Did not solve

#include "bits/stdc++.h"
using namespace std;
#define all(x) (x).begin(),(x).end()
int main()
{
	ios::sync_with_stdio(0); cin.tie(0);
	int T; cin >> T;
	while (T--)
	{
		int n, i;
		cin >> n;
		vector<int> l(n), r(n), a(n);
		for (i = 0; i < n; i++) cin >> l[i] >> r[i];
		vector e(n, vector<int>(0));
		for (i = 1; i < n; i++)
		{
			int u, v;
			cin >> u >> v;
			--u, --v;
			e[u].push_back(v);
			e[v].push_back(u);
		}
		long long ans = 0;
		function<void(int)> dfs = [&](int u) {
			int mx = l[u];
			for (int v : e[u])
			{
				erase(e[v], u);
				dfs(v);
				mx = max(mx, a[v]);
			}
			a[u] = min(mx, r[u]);
			for (int v : e[u]) ans += max(0, a[v] - a[u]);
		};
		dfs(0);
		cout << ans + a[0] << "\n";
	}
}

for _ in range(int(input())):
	n = int(input())
	l,r=[],[]
	for i in range(n):
		L,R=map(int,input().split())
		l.append(L)
		r.append(R)
	e=[[] for i in range(n)]
	for i in range(n-1):
		u,v=map(int,input().split())
		e[u-1].append(v-1)
		e[v-1].append(u-1)
	stk=[0]
	f=[-1]*n
	nfd=[]
	while stk:
		u=stk.pop()
		nfd.append(u)
		for v in e[u]:
			if v!=f[u]:
				f[v]=u
				stk.append(v)
	nfd.reverse()
	ans=0
	for u in nfd:
		for v in e[u]:
			if f[v]==u:
				l[u]=max(l[u],l[v])
		l[u]=min(l[u],r[u])
		for v in e[u]:
			if f[v]==u:
				ans+=max(0,l[v]-l[u])
	print(ans+l[0])

Good problem

Average problem

Bad problem

Did not solve

#include "bits/stdc++.h"
using namespace std;
const int N=1e6+5;
vector<int> e[N];
int w[N],dfn[N],nfd[N],pre[N],suf[N],low[N];
bool ed[N];
int id;
void dfs(int u)
{
	if (ed[u]) return;
	ed[u]=1;
	dfn[u]=++id;
	nfd[id]=u;
	for (int v:e[u]) dfs(v);
	ed[u]=0;
	low[u]=id;
}
int main()
{
	ios::sync_with_stdio(0); cin.tie(0);
	cout<<fixed<<setprecision(15);
	int T; cin>>T;
	while (T--)
	{
		int n,m,i,j;
		cin>>n;
		for (i=1; i<=n; i++)
		{
			e[i].clear();
			id=0;
		}
		for (i=1; i<=n; i++) cin>>w[i];
		for (i=1; i<n; i++)
		{
			int u,v;
			cin>>u>>v;
			e[u].push_back(v);
			e[v].push_back(u);
		}
		dfs(1);
		for (i=1; i<=n; i++) pre[i]=max(pre[i-1],w[nfd[i]]);
		suf[n+1]=0;
		for (i=n; i; i--) suf[i]=max(suf[i+1],w[nfd[i]]);
		int mx=0;
		for (i=1; i<=n; i++) if (max(pre[dfn[i]-1],suf[low[i]+1])>w[i]&&w[i]>w[mx]) mx=i;
		cout<<mx<<'\n';
	}
}

for _ in range(int(input())):
	n = int(input())
	w=[int(x) for x in input().split()]
	wb=[[] for i in range(n)]
	for i in range(n):
		w[i]-=1
		wb[w[i]].append(i)
	e=[[] for i in range(n)]
	for i in range(n-1):
		u,v=map(int,input().split())
		e[u-1].append(v-1)
		e[v-1].append(u-1)
	stk=[0]
	f,dfn=[-1]*n,[0]*n
	nfd=[]
	while stk:
		u=stk.pop()
		dfn[u]=len(nfd)
		nfd.append(u)
		for v in e[u]:
			if v!=f[u]:
				f[v]=u
				stk.append(v)
	sz=[1]*n
	nfd.reverse()
	for u in nfd:
		if u==0:
			continue
		sz[f[u]]+=sz[u]
	wb.reverse()
	mx=-1;mn=n+1
	for v in wb:
		for u in v:
			if mn<dfn[u] or mx>dfn[u]+sz[u]-1:
				print(u+1)
				break
		else:
			for u in v:
				mn=min(mn,dfn[u])
				mx=max(mx,dfn[u])
			continue
		break
	else:
		print(0)

Good problem

Average problem

Bad problem

Did not solve

#include "bits/stdc++.h"
using namespace std;
#define all(x) (x).begin(),(x).end()
const int N = 5e5 + 2;
vector<int> e[N];
int dfn[N], nfd[N], low[N];
int id, n;
void dfs(int u)
{
	nfd[dfn[u] = ++id] = u;
	for (int v : e[u]) erase(e[v], u), dfs(v);
	low[u] = id;
}
int main()
{
	ios::sync_with_stdio(0); cin.tie(0);
	cout << fixed << setprecision(15);
	int T; cin >> T;
	while (T--)
	{
		int n, m, i, j;
		cin >> n;
		vector w(n, vector<int>());
		vector<int> c(n + 1), ans;
		for (i = 1; i <= n; i++)
		{
			e[i].clear(); id = 0;
			cin >> j;
			w[j - 1].push_back(i);
		}
		for (i = 1; i < n; i++)
		{
			int u, v;
			cin >> u >> v;
			e[u].push_back(v);
			e[v].push_back(u);
		}
		dfs(1);
		int l = n + 1, r = 0;
		set<int> s;
		for (i = n - 1; i >= 0; i--)
		{
			if (s.size())
			{
				for (int u : w[i])
				{
					int mx = 0;
					for (j = dfn[u] - 1; j; j -= j & -j) mx = max(mx, c[j]);
					if ((*s.begin() < dfn[u] || *s.rbegin() > low[u]) && mx <= low[u] && dfn[u] <= l && low[u] >= r)
						ans.push_back(u);
				}
				for (int u : w[i])
				{
					int mn = *s.begin(), mx = *s.rbegin();
					int L = dfn[u], R = low[u];
					if (mn >= L && mx <= R) continue;
					if (mn >= L && mn <= R) mn = *s.upper_bound(R);
					if (mx >= L && mx <= R) mx = *prev(s.lower_bound(L));
					auto fun = [&](int x, int y) {
						if (x > y) swap(x, y);
						l = min(l, y), r = max(r, x);
						for (j = x; j <= n; j += j & -j) c[j] = max(c[j], y);
					};
					fun(mn, dfn[u]);
					fun(mx, dfn[u]);
				}
			}
			for (int u : w[i]) s.insert(dfn[u]);
		}
		sort(all(ans));
		cout << ans.size();
		for (int x : ans) cout << ' ' << x;
		cout << '\n';
	}
}

Good problem

Average problem

Bad problem

Did not solve

#include<bits/stdc++.h>
using namespace std;
struct apos{
	long long a;
	long long b;
	friend bool operator<(apos a,apos b){
		return a.b-a.a<b.b-b.a;
	}
}ap[3000];
long long dp[3001][3],tdp[3001][3],ans[3001],inf=1000000000000000000LL;
int main(){
	ios::sync_with_stdio(false),cin.tie(0);
	int T,n,i,j,t;
	for(cin>>T;T>0;T--)
	{
		cin>>n;
		for(i=0;i<n;i++)cin>>ap[i].a>>ap[i].b;
		sort(ap,ap+n);
		for(i=0;i<=n;i++)
		{
			for(j=0;j<3;j++)
			{
				dp[i][j]=inf;
				tdp[i][j]=inf;
			}
			ans[i]=inf;
		}
		for(i=0;i<n;i++)
		{
			tdp[1][0]=min(tdp[1][0],ap[i].a*2);
			tdp[1][1]=min(tdp[1][1],ap[i].a);
			for(j=1;j<n;j++)
			{
				for(t=0;t<3;t++)tdp[j+1][t]=min(tdp[j+1][t],dp[j][t]+ap[i].a+ap[i].b);
				tdp[j+1][1]=min(tdp[j+1][1],dp[j][0]+ap[i].b);
				tdp[j+1][2]=min(tdp[j+1][2],dp[j][1]+ap[i].a);
				ans[j+1]=min(ans[j+1],dp[j][1]+ap[i].b);
				ans[j+1]=min(ans[j+1],dp[j][2]+ap[i].b*2);
			}
			for(j=1;j<=n;j++)
			{
				for(t=0;t<3;t++)
				{
					dp[j][t]=min(dp[j][t],tdp[j][t]);
					tdp[j][t]=inf;
				}
			}
		}
		for(i=2;i<=n;i++)cout<<ans[i]<<' ';
		cout<<'\n';
	}
	return 0;
}

class Q:
	def __init__(this,x,y):
		this.x,this.y=x+y,x-y
	def __lt__(this,rhs):
		return this.y<rhs.y
for _ in range(int(input())):
	n=int(input())
	a=[0]*n
	for i in range(n):
		x,y=map(int,input().split())
		a[i]=Q(x,y)
	a.sort()
	l=[10**18]*(n+1)
	sl=l.copy();slt=l.copy();slrt=l.copy()
	s=10**18
	for m,t in enumerate(a):
		x,y=t.x,t.y
		for i in range(m,0,-1):
			slrt[i+1]=min(slrt[i+1],sl[i]+x+y,slt[i]+x*2+y*2)
			slt[i+1]=min(slt[i+1],slt[i]+x*2,sl[i]+x-y)
			sl[i+1]=min(sl[i+1],sl[i]+x*2,l[i]+x+y)
			l[i+1]=min(l[i+1],l[i]+x*2)
		sl[1]=min(sl[1],x-y)
		l[1]=min(l[1],x*2-y*2)
	print(*[x//2 for x in slrt[2:]])

Good problem

Average problem

Bad problem

Did not solve

#include<bits/stdc++.h>
using namespace std;
struct linex{
	int v;
	int w;
	int nxt;
	int c;
};
int head[210],cnt,cpos[210],vis[210],qvis[210],totc,dis[210],inf=1000000000;
linex l[21000];
queue<int>que;
void add(int u,int v,int w,int c){
	l[cnt].nxt=head[u];
	head[u]=cnt;
	l[cnt].v=v;
	l[cnt].w=w;
	l[cnt].c=c;
	cnt++;
	l[cnt].nxt=head[v];
	head[v]=cnt;
	l[cnt].v=u;
	l[cnt].w=0;
	l[cnt].c=-c;
	cnt++;
}
int spfa(int st,int en,int n){
	int i,j,t;
	for(i=0;i<n;i++)
	{
		vis[i]=0;
		dis[i]=inf;
		cpos[i]=head[i];
	}
	que.push(st);
	dis[st]=0;
	vis[st]=1;
	qvis[st]=1;
	while(!que.empty())
	{
		t=que.front();
		que.pop();
		qvis[t]=0;
		for(j=head[t];j!=-1;j=l[j].nxt)
		{
			if(l[j].w>0&&dis[l[j].v]>dis[t]+l[j].c)
			{
				dis[l[j].v]=dis[t]+l[j].c;
				vis[l[j].v]=1;
				if(!qvis[l[j].v])
				{
					que.push(l[j].v);
					qvis[l[j].v]=1;
				}
			}
		}
	}
	return vis[en];
}
long long dfs(int p,int en,int curr){
	int x,j,flow=0;
	if(p==en)return curr;
	for(j=cpos[p];j!=-1&&flow<curr;j=l[j].nxt)
	{
		cpos[p]=j;
		qvis[p]=1;
		if(l[j].w>0&&dis[l[j].v]==dis[p]+l[j].c&&!qvis[l[j].v])
		{
			x=dfs(l[j].v,en,min(curr-flow,l[j].w));
			flow+=x;
			l[j].w-=x;
			l[j^1].w+=x;
			totc+=l[j].c*x;
		}
		qvis[p]=0;
	}
	return flow;
}
int p[100],q[100],id[100][100];
int cabs(int x){
	if(x<0)x=-x;
	return x;
}
struct point{
	int x;
	int y;
	int tx;
	int ty;
}pc[100];
vector<int>ansv[2];
int main(){
	ios::sync_with_stdio(false),cin.tie(0);
	srand(time(0));
	int T,n,i,j,flow,flag;
	for(cin>>T;T>0;T--)
	{
		cin>>n;
		for(i=0;i<n;i++)
		{
			cin>>p[i];
			p[i]--;
		}
		for(i=0;i<n;i++)
		{
			cin>>q[i];
			q[i]--;
		}
		for(i=0;i<n*2+2;i++)head[i]=-1;
		cnt=0;
		totc=0;
		flow=0;
		for(i=0;i<n;i++)
		{
			add(n*2,i,1,0);
			add(i+n,n*2+1,1,0);
			for(j=0;j<n;j++)
			{
				id[i][j]=cnt;
				add(i,j+n,1,cabs(i-j)+cabs(p[i]-q[j]));
			}
		}
		while(spfa(n*2,n*2+1,n*2+2))flow+=dfs(n*2,n*2+1,inf);
		for(i=0;i<n;i++)
		{
			pc[i].x=i;
			pc[i].y=p[i];
			for(j=0;j<n;j++)
			{
				if(l[id[i][j]].w==0)
				{
					pc[i].tx=j;
					pc[i].ty=q[j];
				}
			}
		}
		while(1)
		{
			flag=0;
			for(i=0;i<n;i++)
			{
				for(j=0;j<n;j++)
				{
					if(pc[j].tx<=pc[i].x&&pc[i].x<pc[j].x&&pc[j].x<=pc[i].tx)
					{
						ansv[0].push_back(pc[i].x);
						ansv[1].push_back(pc[j].x);
						swap(pc[i].x,pc[j].x);
						flag=1;
					}
				}
			}
			if(!flag)break;
		}
		while(1)
		{
			flag=0;
			for(i=0;i<n;i++)
			{
				for(j=0;j<n;j++)
				{
					if(pc[j].ty<=pc[i].y&&pc[i].y<pc[j].y&&pc[j].y<=pc[i].ty)
					{
						ansv[0].push_back(pc[i].x);
						ansv[1].push_back(pc[j].x);
						swap(pc[i].y,pc[j].y);
						flag=1;
					}
				}
			}
			if(!flag)break;
		}
		cout<<ansv[0].size()<<'\n';
		for(i=0;i<ansv[0].size();i++)cout<<ansv[0][i]+1<<' '<<ansv[1][i]+1<<'\n';
		ansv[0].clear();
		ansv[1].clear();
	}
	return 0;
}

Good problem

Average problem

Bad problem

Did not solve

#include<bits/stdc++.h>
using namespace std;
const long long mod=1000000007;
string s;
long long dp[15][15][1<<14],vl[200],sgvl[15][15][15][15];
long long dp2[15][1<<14],dpvl2[15][1<<14];
int psum[15][15];
long long getvl(long long l,long long r,long long x,long long y){
	if(l>=r||x>=y)return 0;
	return vl[psum[r][y]-psum[l][y]-psum[r][x]+psum[l][x]];
}
long long f(long long l,long long r,long long x,long long y){
	long long ans=0,i,c,tl,tr,tx,ty;
	for(i=0;i<16;i++)
	{
		c=0;
		tl=l;
		tr=r;
		tx=x;
		ty=y;
		if(i&1)
		{
			c^=1;
			tl++;
		}
		if(i&2)
		{
			c^=1;
			tr--;
		}
		if(i&4)
		{
			c^=1;
			tx++;
		}
		if(i&8)
		{
			c^=1;
			ty--;
		}
		if(c)ans=(ans+mod-getvl(tl,tr,tx,ty))%mod;
		else ans=(ans+getvl(tl,tr,tx,ty))%mod;
	}
	return ans;
}
int main(){
	ios::sync_with_stdio(false),cin.tie(0);
	int T,n,i,j,t,p,l,r,len,cmsk;
	long long ans;
	for(cin>>T;T>0;T--)
	{
		cin>>n;
		vl[0]=0;
		for(i=1;i<=n*n;i++)vl[i]=(vl[i-1]*2+1)%mod;
		for(i=0;i<=n;i++)
		{
			for(j=0;j<=n;j++)psum[i][j]=0;
		}
		for(i=0;i<n;i++)
		{
			cin>>s;
			for(j=0;j<n;j++)psum[i+1][j+1]=s[j]-'0';
		}
		for(i=0;i<n;i++)
		{
			for(j=0;j<=n;j++)psum[i+1][j]+=psum[i][j];
		}
		for(i=0;i<=n;i++)
		{
			for(j=0;j<n;j++)psum[i][j+1]+=psum[i][j];
		}
		for(l=0;l<=n;l++)
		{
			for(r=l+1;r<=n;r++)
			{
				for(i=0;i<=n;i++)
				{
					for(j=i+1;j<=n;j++)sgvl[l][r][i][j]=f(l,r,i,j);
				}
			}
		}
		for(i=0;i<=n;i++)
		{
			for(j=0;j<=n;j++)
			{
				for(t=0;t<(1<<n);t++)dp[i][j][t]=0;
			}
		}
		for(i=0;i<=n;i++)dp[i][i][0]=1;
		for(len=1;len<=n;len++)
		{
			for(l=0;l+len<=n;l++)
			{
				r=l+len;
				for(i=l+1;i<r;i++)
				{
					for(j=1;j<(1<<n);j++)
					{
						for(t=0;t<n;t++)
						{
							cmsk=j;
							for(p=t;p<n&&(j>>p&1^1);p++)
							{
								cmsk|=(1<<p);
								dp[l][r][cmsk]=(dp[l][r][cmsk]+dp[l][i][j]*sgvl[i][r][t][p+1])%mod;
							}
						}
					}
				}
				for(j=1;j<(1<<n);j++)
				{
					for(i=0;(j>>i&1^1);i++);
					for(t=n-1;(j>>t&1^1);t--);
					sgvl[l][r][i][t+1]=(sgvl[l][r][i][t+1]+mod-dp[l][r][j])%mod;
				}
				for(i=0;i<n;i++)
				{
					cmsk=0;
					for(t=i;t<n;t++)
					{
						cmsk|=(1<<t);
						dp[l][r][cmsk]=(dp[l][r][cmsk]+sgvl[l][r][i][t+1])%mod;
					}
				}
				if(len==1)continue;
				for(j=0;j<(1<<n);j++)dp[l][r][j]=(dp[l][r-1][j]+dp[l][r][j])%mod;
			}
		}
		for(i=0;i<=n;i++)
		{
			for(j=0;j<(1<<n);j++)
			{
				dp2[i][j]=0;
				dpvl2[i][j]=0;
			}
		}
		dp2[0][0]=1;
		for(l=0;l<n;l++)
		{
			for(j=0;j<(1<<n);j++)
			{
				dp2[l+1][j]=(dp2[l+1][j]+dp2[l][j])%mod;
				dpvl2[l+1][j]=(dpvl2[l+1][j]+dpvl2[l][j])%mod;
			}
			for(r=l+1;r<=n;r++)
			{
				for(j=0;j<(1<<n);j++)
				{
					for(i=0;i<n;i++)
					{
						cmsk=j;
						for(t=i;t<n&&(j>>t&1^1);t++)
						{
							cmsk|=(1<<t);
							dp2[r][cmsk]=(dp2[r][cmsk]+dp2[l][j]*sgvl[l][r][i][t+1])%mod;
							dpvl2[r][cmsk]=(dpvl2[r][cmsk]+(dp2[l][j]+dpvl2[l][j])*sgvl[l][r][i][t+1])%mod;
						}
					}
				}
			}
		}
		ans=0;
		for(j=1;j<(1<<n);j++)ans=(ans+dpvl2[n][j])%mod;
		cout<<ans<<'\n';
	}
	return 0;
}

Well, we gave you enough hints in the examples. I believe you can already guess it.

for i in range(int(input())):
    x,y=map(int,input().split())
    if x==y==1:
        print(1)
    else:
        print(y-x)

Assume that you chose some two indices $$$i$$$ and $$$j$$$ included in the subsequence, such that $$$i<l$$$ and $$$j>r$$$. Does this actually help decrease the answer in any way?

import sys
input=lambda:sys.stdin.readline().rstrip()
 
for _ in range(int(input())):
    n,l,r=map(int,input().split());l-=1
    arr=[*map(int,input().split())]
    brr=arr[:l]+sorted(arr[l:])
    crr=sorted(arr[:r])[::-1]+arr[r:]
    print(min(sum(brr[l:r]),sum(crr[l:r])))

The answer after choosing two vertices is decided by the degrees of the two vertices, and whether the two vertices are adjacent. Can you find a clean solution where you don't have to care about the latter?

Sure, you can do DP on tree, but that's not what we want you to use. Think easier.

import sys
input=lambda:sys.stdin.readline().rstrip()
for i in range(int(input())):
    n=int(input())
    deg=[0]*n
    adj=[[] for i in range(n)]
    for i in range(n-1):
        u,v=map(int,input().split())
        u-=1;v-=1
        deg[u]+=1
        deg[v]+=1
        adj[u].append(v)
        adj[v].append(u)
    ans=1
    mans=0
    sdeg=sorted(deg)
    for i in range(n):
        ans=deg[i]
        ideg=[]
        for v in adj[i]:
            ideg.append(deg[v])
        ideg.append(deg[i])
        ideg.sort(reverse=True)
        rem=[]
        mx=-1
        for d in ideg:
            if sdeg[-1]==d:
                sdeg.pop()
                rem.append(d)
        rem.reverse()
        if sdeg:
            mx=max(mx,sdeg[-1])
        for v in adj[i]:
            mx=max(mx,deg[v]-1)
        for d in rem:
            sdeg.append(d)
        mans=max(ans+mx-1,mans)
    print(mans)

The maximum score $$$g(x_a,x_b)$$$ after performing $$$x_a$$$ times with two points on $$$y=0$$$, and $$$x_b$$$ times with two points on $$$y=2$$$, can be easily computed in $$$\mathcal{O}(1)$$$ time. But naively checking all possible number of operations takes $$$\mathcal{O}(nm)$$$ time. Is there a property we can abuse so that we can solve for each $$$k$$$ faster?

$$$g(x_a,x_b)$$$ is a function consisting of two prefix sums, each for a strictly decreasing sequence.

#include<bits/stdc++.h>
using namespace std;
using ll=long long;
 
int main()
{
  cin.tie(0)->sync_with_stdio(0);
  int t;cin>>t;
  while(t--)
  {
    int n,m;cin>>n>>m;
    vector<ll>arr(n),brr(m);
    for(ll&i:arr)cin>>i;
    for(ll&i:brr)cin>>i;
    sort(begin(arr),end(arr));
    sort(begin(brr),end(brr));
    vector<ll>asum(n+2),bsum(m+2);
    for(int i=1;i<=n;i++)asum[i]=asum[i-1]+(arr[n-i]-arr[i-1]);
    for(int i=1;i<=m;i++)bsum[i]=bsum[i-1]+(brr[m-i]-brr[i-1]);
    vector<ll>ans{0};
    // maximize asum[ka]+bsum[kb]
    // s.t. ka+kb = x
    //      ka*2+kb <= n -> ka*2+(x-ka) <= n -> ka+x <= n   -> ka <= n-x
    //      ka+kb*2 <= m -> ka+2*(x-ka) <= m -> 2*x-ka <= m -> ka >= 2*x-m
    //      ka >= 0, x-ka >= 0
    for(int x=1;2*x-m<=n-x;x++)
    {
      ll L=max(0,2*x-m),R=min(x,n-x);
      if(L>R)break;
      auto f=[&](int ka){return asum[ka]+bsum[x-ka];};
      while(R-L>3)
      {
        ll mL=(L*2+R)/3,mR=(L+R*2)/3;
        if(f(mL)>f(mR))R=mR;
        else L=mL;
      }
      ll mans=0;
      for(int i=L;i<=R;i++)
      {
        mans=max(mans,f(i));
      }
      ans.push_back(mans);
    }
    int kmax=(int)size(ans)-1;
    cout<<kmax<<"\n";
    for(int i=1;i<=kmax;i++)cout<<ans[i]<<" \n"[i==kmax];
  }
}

When $$$(u,v)$$$ is a good pair, $$$f(u,v)$$$ is actually simply represented as follows.

Naively calculating this for all good pairs takes $$$\mathcal{O}(n^2)$$$ time. Can you find a different representation where we can count wiser?

#include<bits/stdc++.h>
using namespace std;
using ll=long long;

int main()
{
    cin.tie(0)->sync_with_stdio(0);
    int t;cin>>t;
    while(t--)
    {
        ll n;cin>>n;
        vector<ll>d(n,0),s(n),dc(n),dcs;
        vector<vector<ll>>adj(n);
        for(int i=1;i<n;i++)
        {
            int u,v;cin>>u>>v;
            u--;v--;
            adj[u].push_back(v);
            adj[v].push_back(u);
        }
        auto dfs1=[&](auto dfs1,int v,int p=-1)->void
        {
            dc[d[v]]++;
            ll sz=1;
            for(int w:adj[v])if(w!=p)
            {
                d[w]=d[v]+1;
                dfs1(dfs1,w,v);
                sz+=s[w];
            }
            s[v]=sz;
        };
        dfs1(dfs1,0);
        dcs=dc;
        for(int i=n-2;i>=0;i--)dcs[i]+=dcs[i+1];
        ll ans=0,ans2=0;
        auto dfs2=[&](auto dfs2,int v,int p=-1)->void
        {
            // v is min
            ans+=2*d[v]*(dcs[d[v]]-s[v]);
            // v is lca
            ll subcnt=s[v]-1,lcnt=0;
            for(int w:adj[v])if(w!=p)
            {
                lcnt+=(subcnt-s[w])*s[w];
                dfs2(dfs2,w,v);
            }
            ans2+=(2*d[v]+1)*(lcnt/2);
        };
        dfs2(dfs2,0);
        for(int i=0;i<n;i++)
        {
            ans2+=i*dc[i]*(dc[i]-1);
        }
        cout<<ans-ans2<<"\n";
    }
}

The subproblem where you find the number of different balanced bracket sequences of length $$$2k$$$ is already well-known.

Now recall everyone's favorite data structure to solve problems related to balanced bracket sequences — the stack!

#include<bits/stdc++.h>
using namespace std;
using ll=long long;
const ll md=998244353;
 
int main()
{
    int t;cin>>t;
    vector<ll>ctl(5050);
    ctl[0]=1;
    for(int n=1;n<5050;n++)
    {
        for(int i=1;i<=n;i++)
        {
            ctl[n]=(ctl[n]+ctl[i-1]*ctl[n-i]%md)%md;
        }
    }
    while(t--)
    {
        int n;cin>>n;
        ll ans=ctl[n];
        cout<<ans<<" ";
        string s(2*n+2,'.');
        s[0]='(';s[2*n+1]=')';
        for(int a=0;a<n;a++)
        {
            int i,j;cin>>i>>j;
            ans=1;
            s[i]='(';
            s[j]=')';
            string stk;
            for(char c:s)
            {
                if(c==')')
                {
                    int cnt=0;
                    while(stk.back()!='(')
                    {
                        cnt++;
                        stk.pop_back();
                    }
                    stk.pop_back();
                    ans=(ans*ctl[cnt/2])%md;
                }
                else stk+=c;
            }
            cout<<ans<<" \n"[a+1==n];
        }
    }
}

In the solution of the easy subtask, we manually enumerated the minimal balanced subsequences, each time in $$$\mathcal{O}(n)$$$.

You can show that most minimal balanced subsequences do not change when a new clue is added. What data structure will you choose in order to utilize this fact as much as possible?

#include<bits/stdc++.h>
using namespace std;
using ll=long long;
const ll nil=-1;
const ll md=998244353;
 
ll pw(ll a,ll b)
{
    ll c=1;
    while(b>0)
    {
        if(b&1)c=c*a%md;
        a=a*a%md;
        b>>=1;
    }
    return c;
}
 
int main()
{
    cin.tie(0)->sync_with_stdio(0);
    vector<ll>fac(1010101,1),ctl(505050);
    for(int i=1;i<1010101;i++)fac[i]=fac[i-1]*i%md;
    for(int i=0;i<505050;i++)
    {
        ctl[i]=fac[i*2]*pw(fac[i]*fac[i+1]%md,md-2)%md;
    }
    ll t;cin>>t;
    while(t--)
    {
        ll n;cin>>n;
        vector<ll>L(n*2,nil),R(n*2,nil),P(n*2,nil),S(n*2);
        auto update=[&](ll x)
        {
            S[x]=1;
            if(L[x]!=nil)S[x]+=S[L[x]];
            if(R[x]!=nil)S[x]+=S[R[x]];
        };
        auto rotate=[&](ll x)
        {
            ll dummy;
            ll p=P[x];
            ll b=nil;
            if(p==nil)return;
            if(x==L[p])
            {
                L[p]=b=R[x];
                R[x]=p;
            }
            else
            {
                R[p]=b=L[x];
                L[x]=p;
            }
            P[x]=P[p];P[p]=x;
            if(b!=nil)P[b]=p;
            (P[x]!=nil?p==L[P[x]]?L[P[x]]:R[P[x]]:dummy)=x;
            update(p);update(x);
        };
        auto splay=[&](ll x)
        {
            while(P[x]!=nil)
            {
                ll p=P[x];
                ll g=P[p];
                if(g!=nil)
                {
                    if((x==L[p])==(p==L[g]))rotate(p);
                    else rotate(x);
                }
                rotate(x);
            }
        };
        update(0);
        for(int i=1;i<n*2;i++)
        {
            L[i]=i-1;
            P[i-1]=i;
            update(i);
        }
        splay(0);
        ll ans=ctl[n];
        cout<<ans<<" ";
        for(int i=0;i<n;i++)
        {
            ll li,ri;cin>>li>>ri;
            //li=(li+ans%(2*n))%(2*n);
            //ri=(ri+ans%(2*n))%(2*n);
            li--;ri--;
            splay(li);
            ans=ans*pw(ctl[S[li]/2],md-2)%md;
            ll lli=L[li];
            if(L[li]!=nil)P[L[li]]=nil;
            if(R[li]!=nil)P[R[li]]=nil;
            L[li]=nil;
            R[li]=nil;
            update(li);
            // now L[li] separated from tree
            // should not be affected in splay ri
            splay(ri);
            ll lri=L[ri];
            ll rri=R[ri];
            if(L[ri]!=nil)P[L[ri]]=nil;
            if(R[ri]!=nil)P[R[ri]]=nil;
            L[ri]=nil;
            R[ri]=nil;
            update(ri);
            if(lri!=nil)
            {
                splay(lri);
                ans=ans*ctl[S[lri]/2]%md;
            }
            if(lli!=nil&&rri!=nil)
            {
                while(L[rri]!=nil)rri=L[rri];
                splay(lli);
                splay(rri);
                P[lli]=rri;
                L[rri]=lli;
                update(rri);
                ans=ans*ctl[S[rri]/2]%md;
            }
            else if(lli!=nil||rri!=nil)
            {
                if(lli!=nil)
                {
                    splay(lli);
                    ans=ans*ctl[S[lli]/2]%md;
                }
                else
                {
                    splay(rri);
                    ans=ans*ctl[S[rri]/2]%md;
                }
            }
            L[ri]=li;
            P[li]=ri;
            update(ri);
            // make dummy tree (for invariant maintaining)
            cout<<ans<<" \n"[i+1==n];
        }
    }
}

Consider maintaining the MBSes directly, but instead of using a complex data structure, we will use a small-to-large trick with linked lists. Precisely, every time we have to split a linked list, we will identify the smaller section and split it out in $$$\mathcal{O}(|small|)$$$ time, and then the amortized time complexity will be $$$\mathcal{O}(n \log n)$$$. In this solution, we do not consider bracket pairs as MBSes, for a specific reason.

Given a linked list of indices and two nodes on it, we can identify the smaller section in $$$\mathcal{O}(|small|)$$$ by iterating over both lists at the same time, interlacing the operations. The list which hit the end earlier will be the smaller one, and then we immediately know the value of $$$|small|$$$.

The issue is that you cannot identify the size of both lists in $$$\mathcal{O}(|small|)$$$ time. This is hard to fix using only a linked list. Instead, we will also maintain the implicit rooted tree structure of the bracket sequence. The MBSes correspond to vertices, and the bracket pairs correspond to edges. Initially, the tree just consists of one vertex of $$$2n$$$ brackets.

Now, for each list node, we add a pointer to the corresponding tree vertex. Now we can know which tree vertex the list node corresponds to immediately, and if we bookkeep size informations in the tree vertices, we can also find the sum of two list sizes in $$$\mathcal{O}(1)$$$. Therefore we can now know the size of both lists in $$$\mathcal{O}(|small|)$$$ time.

The only issue is with how to maintain the pointers to the tree vertices. But no problem, you can just do the small-to-large for this also. After splitting the tree vertex into two vertices and one edge, redirect the smaller list to the new smaller vertex. Theoretically there are enough ways to do this, such as swapping vertex indices. Also it is notable that it is not necessary to maintain the whole tree in this process, it is quite sufficient to just maintain it implicitly by just maintaining a sequence of sizes.

The problem is solved online with $$$\mathcal{O}(n \log n)$$$ amortized time complexity.

#pragma GCC optimize("O3,unroll-loops")

#include <iostream>
#include <numeric>
#include <vector>

constexpr int MAX_N = 3e5;
constexpr long long MOD = 998244353;

struct _inv_small {
  long long data[MAX_N + 2] = {0};
  constexpr _inv_small() {
    data[1] = 1;
    for (int i = 2; i <= MAX_N; i += 2) {
      data[i] = MOD - (MOD / i) * data[MOD % i] % MOD;
      data[i + 1] = MOD - (MOD / (i + 1)) * data[MOD % (i + 1)] % MOD;
    }
  }
};

constexpr _inv_small __inv_small;
#define inv_small(x) __inv_small.data[x]

#define inv(x) (x < MAX_N ? inv_small(x) : data[(x) - MAX_N])
struct _inv {
  long long data[MAX_N + 3] = {0};
  constexpr _inv() {
    for (int i = 0; i <= MAX_N + 1; i += 2) {
      data[i] = MOD - (MOD / (i + MAX_N)) * inv(MOD % (i + MAX_N)) % MOD;
      data[i + 1] = MOD - (MOD / (i + MAX_N + 1)) * inv(MOD % (i + MAX_N + 1)) % MOD;
    }
  }
};
#undef inv

constexpr _inv __inv;
#define inv(x) (x < MAX_N ? inv_small(x) : __inv.data[(x) - MAX_N])

struct _catalan {
  long long data[MAX_N + 2] = {1};
  constexpr _catalan() {
    for (int i = 1; i <= MAX_N; i += 2) {
      data[i] = (4 * i - 2) * data[i - 1] % MOD * inv(i + 1) % MOD;
      data[i + 1] = (4 * i + 2) * data[i] % MOD * inv(i + 2) % MOD;
    }
  }
};

constexpr _catalan __catalan;
#define catalan(x) __catalan.data[x]

struct _catalan_inv {
  long long data[MAX_N + 2] = {1};
  constexpr _catalan_inv() {
    for (int i = 1; i <= MAX_N; i += 2) {
      data[i] = inv(2) * inv(2 * i - 1) % MOD * data[i - 1] % MOD * (i + 1) % MOD;
      data[i + 1] = inv(2) * inv(2 * i + 1) % MOD * data[i] % MOD * (i + 2) % MOD;
    }
  }
};

constexpr _catalan_inv __catalan_inv;
#define catalan_inv(x) __catalan_inv.data[x]

int main() {
  std::cin.tie(0)->sync_with_stdio(0);

  int t;
  std::cin >> t;

  for (int _ = 0; _ < t; ++_) {
    int n;
    std::cin >> n;

    std::vector<bool> used(2 * n + 2);

    std::vector<int> point(2 * n + 2);
    std::iota(point.begin(), point.end(), 1);

    std::vector<int> size(2 * n + 2);

    std::vector<int> parent = {0};
    std::vector<int> parent_idx(2 * n + 2);

    used[0] = true;
    point[0] = 2 * n + 2, point[2 * n + 1] = 2 * n + 1;
    size[0] = 2 * n;

    long long ans = catalan(n);

    std::cout << ans << " ";

    for (int i = 0, l, r; i < n; ++i) {
      std::cin >> l >> r;

      point[l] = r + 1, point[r] = r;
      used[l] = true;

      int p = parent[parent_idx[l]];

      int out_ptr = p + 1, ins_ptr = l + 1;
      int out_size = 0, ins_size = 0;

      while (point[out_ptr] != out_ptr && point[ins_ptr] != ins_ptr) {
        out_size += !used[out_ptr];
        ins_size += !used[ins_ptr];
        out_ptr = point[out_ptr];
        ins_ptr = point[ins_ptr];
      }

      ans = (ans * catalan_inv(size[p] / 2)) % MOD;

      int upd_ptr;
      if (point[out_ptr] == out_ptr) {
        upd_ptr = p + 1;
        parent.push_back(p);
        parent[parent_idx[l]] = l;

        ins_size = size[p] - 2 - out_size;
        size[p] = out_size, size[l] = ins_size;
      } else {
        upd_ptr = l + 1;
        parent.push_back(l);

        out_size = size[p] - 2 - ins_size;
        size[p] = out_size, size[l] = ins_size;
      }

      ans = (ans * catalan(size[p] / 2)) % MOD * catalan(size[l] / 2) % MOD;

      while (point[upd_ptr] != upd_ptr) {
        parent_idx[upd_ptr] = parent.size() - 1;
        upd_ptr = point[upd_ptr];
      }

      std::cout << ans << " ";
    }

    std::cout << '\n';
  }
}