U397690 斗地主

这题是我上传的 ，自己写的感觉太长，有没有更短一点的

#include<bits/stdc++.h> 
using namespace std;
int main(){
	int n,a[5]={0},w,d,b;
	cin>>n;
	for(int i=1;i<=n;i++){
	cin>>w>>d>>b;
	if(b-1){
	a[w]-=3*d;
	if(w==1){a[2]+=d;a[3]+=d;a[4]+=d;}
    else if(w==2){a[1]+=d;a[3]+=d;a[4]+=d;}
	else if(w==3){a[1]+=d;a[2]+=d;a[4]+=d;}
	else{a[1]+=d;a[2]+=d;a[3]+=d;}
	}
	else{
	a[w]+=3*d;
	if(w==1){a[2]-=d;a[3]-=d;a[4]-=d;}
    else if(w==2){a[1]-=d;a[3]-=d;a[4]-=d;}
	else if(w==3){a[1]-=d;a[2]-=d;a[4]-=d;}
	else{a[1]-=d;a[2]-=d;a[3]-=d;}
	}
	}
	cout<<a[1]<<' '<<a[2]<<' '<<a[3]<<' '<<a[4];
	return 0; 
}