1266E - Spaceship Solitaire - CodeForces Solution

#include<bits/stdc++.h>
using namespace std;
int n;
long long a[200001];
map<pair<long long,long long>,long long> q;
long long sum=0;
int main(){
	scanf("%d",&n);
	for(int i=1;i<=n;++i){
		scanf("%lld",a+i);
		sum+=a[i];
	}
	int T;scanf("%d",&T);
	while(T--){
		long long s,t,u;scanf("%lld%lld%lld",&s,&t,&u);
		if(q[{s,t}]!=0){
			long long cnt=a[q[{s,t}]];
			a[q[{s,t}]]++;
			q[{s,t}]=0;
			if(cnt>=0) ++sum; 
		}
		if(u!=0){
			q[{s,t}]=u;
			long long cnt=a[u];
			a[u]--;
			if(cnt>0) --sum; 
		}
		printf("%lld\n",sum);
	}
	return 0;
}