-
Notifications
You must be signed in to change notification settings - Fork 4
/
Copy pathG.cpp
119 lines (108 loc) · 2.44 KB
/
G.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
#include<bits/stdc++.h>
#define For(i,a,b) for(int i=(a);i<=(b);++i)
#define Rep(i,a,b) for(int i=(a);i>=(b);--i)
#define int long long
using namespace std;
inline int read()
{
char c=getchar();int x=0;bool f=0;
for(;!isdigit(c);c=getchar())f^=!(c^45);
for(;isdigit(c);c=getchar())x=(x<<1)+(x<<3)+(c^48);
return f?-x:x;
}
#define fi first
#define se second
#define pb push_back
#define mkp make_pair
typedef pair<int,int>pii;
typedef vector<int>vi;
#define maxn 1000005
#define inf 0x3f3f3f3f
int n,a[3];
vector<pii>s[3];
int bound[3][2];
mt19937_64 rnd(1919810);
int cnt[3],mid[3];
int res;
int tp[3];
int fa[maxn];
int gf(int x){
while(x^fa[x])x=fa[x]=fa[fa[x]];
return x;
}
void check()
{
For(i,0,2) cnt[i]=0,tp[i]=s[i].size();
For(i,0,n+1) fa[i]=i;
int sum=0; res=0;
while(sum<a[0]+a[1]+a[2]){
int t=-1,mx=-(1ll<<63);
For(i,0,2)
if(tp[i] && s[i][tp[i]-1].se+mid[i]>mx)
mx=s[i][tp[i]-1].se+mid[i],t=i;
if(t==-1)break;
int d=s[t][tp[t]-1].fi; --tp[t];
// cout<<"d: "<<d<<" "<<t<<" "<<tp[t]<<"\n";
d=gf(n-d);
if(d<n) fa[d]=d+1,++cnt[t],++sum,res+=((mx-mid[t])>>25);
}
// cout<<"cnt: ";For(i,0,2)cout<<cnt[i]<<" ";puts("");
}
int work()
{
n=read();
For(i,0,2)s[i].clear();
For(i,0,2)a[i]=read();
int V=n*200+1;
For(i,0,n-1){
int r=read(),t=read(),d=read();
r=(r<<25)|(rnd()%V);
s[t-1].pb(mkp(d,r));
}
For(i,0,2)sort(s[i].begin(),s[i].end(),[](pii a,pii b){return a.se<b.se;});
if(!a[0]&&!a[1]&&!a[2])return 0;
For(i,0,2) bound[i][1]=(1ll<<56),bound[i][0]=-bound[i][1];
while(1){
int r=0;
For(i,0,2) mid[i]=0;
For(i,0,2)For(a,0,1)For(b,0,1){
int j=(i+1)%3,k=(j+1)%3;
int x=bound[j][a],y=-bound[k][b];
if(bound[i][0]<=y-x && y-x<=bound[i][1]) mid[j]+=y,mid[k]+=x,++r;
}
For(i,0,2) mid[i]/=r;
For(i,0,2){
int j=(i+1)%3,k=(j+1)%3,t=mid[j]-mid[k];
if(bound[i][0]>t || t>bound[i][1])return -1;
}
check();
if(cnt[0]+cnt[1]+cnt[2]!=a[0]+a[1]+a[2])return -1;
if(cnt[0]==a[0] && cnt[1]==a[1] && cnt[2]==a[2])return res;
For(i,0,2){
int j=(i+1)%3,k=(j+1)%3;
if(cnt[j]>a[j] && cnt[k]<a[k]) bound[i][1]=min(bound[i][1],mid[j]-mid[k]-1);
if(cnt[j]<a[j] && cnt[k]>a[k]) bound[i][0]=max(bound[i][0],mid[j]-mid[k]+1);
}
}
}
signed main()
{
int T=read();
while(T--)cout<<work()<<"\n";
return 0;
}
/*
12 2 2 2
3 1 2
3 1 2
3 2 2
3 2 2
2 2 4
2 2 4
2 3 4
2 3 4
1 3 6
1 3 6
0 1 6
0 1 6
*/