QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#90790 | #5828. 游戏 | SoyTony | 70 | 975ms | 85128kb | C++14 | 6.3kb | 2023-03-25 12:02:33 | 2023-03-25 12:02:35 |
Judging History
answer
#include<bits/stdc++.h>
using namespace std;
typedef pair<int,int> pii;
const int maxn=2e5+10;
#define val first
#define id second
inline int read(){
int x=0,w=1;char c=getchar();
while(c<'0'||c>'9'){if(c=='-')w=-1;c=getchar();}
while(c<='9'&&c>='0'){x=(x<<3)+(x<<1)+c-'0';c=getchar();}
return x*w;
}
int n,m;
vector<int> E[maxn];
int fa[maxn],dep[maxn],siz[maxn],son[maxn],F[maxn][19];
int top[maxn],dfn[maxn],dfncnt;
pii mx[maxn][3];
vector<int> V;
void dfs1(int u,int f,int d){
fa[u]=f,dep[u]=d,siz[u]=1,mx[u][0]=make_pair(0,u);
F[u][0]=fa[u];
for(int k=1;(1<<k)<=dep[u];++k) F[u][k]=F[F[u][k-1]][k-1];
int maxson=-1;
for(int v:E[u]){
if(v==f) continue;
dfs1(v,u,d+1);
siz[u]+=siz[v];
if(siz[v]>maxson) maxson=siz[v],son[u]=v;
if(mx[v][0].val+1>mx[u][0].val){
mx[u][2]=mx[u][1],mx[u][1]=mx[u][0],mx[u][0]=make_pair(mx[v][0].val+1,mx[v][0].id);
}
else if(mx[v][0].val+1>mx[u][1].val){
mx[u][2]=mx[u][1],mx[u][1]=make_pair(mx[v][0].val+1,mx[v][0].id);
}
else if(mx[v][0].val+1>mx[u][2].val){
mx[u][2]=make_pair(mx[v][0].val+1,mx[v][0].id);
}
}
}
void dfs2(int u,int t){
top[u]=t,dfn[u]=++dfncnt;
if(!son[u]) return;
dfs2(son[u],t);
for(int v:E[u]){
if(v==fa[u]||v==son[u]) continue;
dfs2(v,v);
}
}
inline int get_kthfa(int u,int k){
for(int i=0;(1<<i)<=k;++i){
if(k&(1<<i)) u=F[u][i];
}
return u;
}
inline int get_LCA(int u,int v){
while(top[u]!=top[v]){
if(dep[top[u]]>dep[top[v]]) swap(u,v);
v=fa[top[v]];
}
if(dep[u]>dep[v]) swap(u,v);
return u;
}
inline int get_dis(int u,int v){
int lca=get_LCA(u,v);
return dep[u]+dep[v]-dep[lca]*2;
}
void get_tri(int u){
for(int v:E[u]){
if(v==fa[u]) continue;
for(int k=0;k<=2;++k){
if(mx[u][k].id==mx[v][0].id) continue;
if(mx[u][k].val+1>mx[v][0].val){
mx[v][2]=mx[v][1],mx[v][1]=mx[v][0],mx[v][0]=make_pair(mx[u][k].val+1,mx[u][k].id);
}
else if(mx[u][k].val+1>mx[v][1].val){
mx[v][2]=mx[v][1],mx[v][1]=make_pair(mx[u][k].val+1,mx[u][k].id);
}
else if(mx[u][k].val+1>mx[v][2].val){
mx[v][2]=make_pair(mx[u][k].val+1,mx[u][k].id);
}
break;
}
get_tri(v);
}
}
struct node{
int a,b,c;
int x,y,z;
bool type,mark;
int ID;
}q[maxn<<1];
int tot;
int ct[maxn];
bool cmp_a(node X,node Y){
if(X.a==Y.a){
if(X.b==Y.b){
if(X.c==Y.c) return X.type<Y.type;
return X.c>Y.c;
}
return X.b>Y.b;
}
return X.a>Y.a;
}
bool cmp_b(node X,node Y){
if(X.b==Y.b){
if(X.c==Y.c) return X.type<Y.type;
return X.c>Y.c;
}
return X.b>Y.b;
}
bool cmp_id(node X,node Y){
if(X.type==Y.type) return X.ID<Y.ID;
return X.type>Y.type;
}
#define mid ((l+r)>>1)
set<pii> S;
void cdq(int l,int r){
if(l==r) return;
cdq(l,mid),cdq(mid+1,r);
int i=l,j=mid+1;
// printf("[%d,%d]\n",l,r);
sort(q+l,q+mid+1,cmp_b),sort(q+mid+1,q+r+1,cmp_b);
while(i<=mid&&j<=r){
// printf("(%d %d)\n",i,j);
if(q[i].b<q[j].b){
if(q[j].type&&!q[j].mark){
auto now=S.lower_bound(make_pair(q[j].c,-1));
if(now!=S.end()){
ct[q[j].ID]=(*now).id;
q[j].mark=1;
}
}
++j;
}
else{
if(!q[i].type) S.insert(make_pair(q[i].c,q[i].ID));
++i;
}
}
while(j<=r){
if(q[j].type&&!q[j].mark){
auto now=S.lower_bound(make_pair(q[j].c,-1));
if(now!=S.end()){
ct[q[j].ID]=(*now).id;
q[j].mark=1;
}
}
++j;
}
for(;i>=l;--i){
if(!q[i].type) S.erase(make_pair(q[i].c,q[i].ID));
}
}
inline int calc(int x,int y,int d){
//get z \in (x,y) (x,z)=d
if(!d) return x;
int lca=get_LCA(x,y);
if(lca==x){
return get_kthfa(y,dep[y]-dep[x]-d);
}
else if(lca==y){
return get_kthfa(x,d);
}
else{
if(d<=dep[x]-dep[lca]) return get_kthfa(x,d);
else return get_kthfa(y,dep[x]+dep[y]-2*dep[lca]-d);
}
}
int main(){
// freopen("game.in","r",stdin);
// freopen("game.out","w",stdout);
n=read();
for(int i=1;i<n;++i){
int u=read(),v=read();
E[u].push_back(v),E[v].push_back(u);
}
dfs1(1,0,0);
dfs2(1,1);
get_tri(1);
for(int i=1;i<=n;++i){
++tot;
q[tot].a=mx[i][0].val,q[tot].b=mx[i][1].val,q[tot].c=mx[i][2].val;
q[tot].type=0,q[tot].ID=i;
}
m=read();
for(int i=1;i<=m;++i){
++tot;
q[tot].x=read(),q[tot].y=read(),q[tot].z=read();
V.clear();
V.push_back((q[tot].x+q[tot].y-q[tot].z)/2);
V.push_back((q[tot].x+q[tot].z-q[tot].y)/2);
V.push_back((q[tot].y+q[tot].z-q[tot].x)/2);
sort(V.begin(),V.end());
q[tot].a=V[2],q[tot].b=V[1],q[tot].c=V[0];
q[tot].type=1,q[tot].mark=0,q[tot].ID=i;
}
// for(int i=1;i<=tot;++i){
// printf("i:%d (%d %d %d %d)\n",i,q[i].a,q[i].b,q[i].c,q[i].ID);
// }
sort(q+1,q+tot+1,cmp_a);
// for(int i=1;i<=tot;++i){
// printf("i:%d (%d %d %d %d)\n",i,q[i].a,q[i].b,q[i].c,q[i].ID);
// }
cdq(1,tot);
// for(int i=1;i<=m;++i) printf("i:%d ct:%d\n",i,ct[i]);
sort(q+1,q+tot+1,cmp_id);
for(int i=1;i<=m;++i){
int now[4];
// printf("abc:%d %d %d\n",q[i].a,q[i].b,q[i].c);
// printf("ct:%d\n",ct[i]);
now[1]=calc(ct[i],mx[ct[i]][0].id,q[i].a);
now[2]=calc(ct[i],mx[ct[i]][1].id,q[i].b);
now[3]=calc(ct[i],mx[ct[i]][2].id,q[i].c);
sort(now+1,now+4);
do{
if(get_dis(now[1],now[2])==q[i].x&&get_dis(now[1],now[3])==q[i].y){
printf("%d %d %d\n",now[1],now[2],now[3]);
break;
}
}while(next_permutation(now+1,now+4));
}
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 10
Accepted
time: 7ms
memory: 8372kb
input:
500 279 196 182 105 400 14 278 217 198 411 379 318 120 421 314 95 437 325 23 433 71 485 192 154 90 224 180 71 328 93 183 101 404 349 223 285 492 100 366 480 29 328 200 448 365 156 226 231 129 167 246 273 171 452 284 6 131 471 229 90 480 48 448 157 240 221 7 36 401 200 255 438 436 110 281 489 388 258...
output:
344 21 37 278 498 343 125 362 21 443 350 176 22 260 64 53 436 17 443 113 19 330 29 311 45 22 278 278 68 157 353 311 396 335 454 127 452 436 361 275 463 361 146 17 21 311 238 396 481 136 217 341 143 152 353 349 270 201 452 269 215 57 401 246 215 437 328 361 266 93 155 281 68 353 437 436 386 346 333 1...
result:
ok Accepted!
Test #2:
score: 10
Accepted
time: 3ms
memory: 8976kb
input:
2000 643 1871 689 23 1822 1443 1031 1027 1655 845 189 771 1561 498 19 1778 576 1080 904 717 1690 291 1387 1077 398 811 1310 1231 645 1291 480 927 330 170 1464 1057 1033 894 1308 288 1292 1529 1212 122 1108 401 89 1118 1058 1088 1764 1314 981 1255 1893 864 180 1887 1903 843 734 1412 883 1013 1739 124...
output:
1118 37 697 755 1685 969 1296 47 220 1110 1210 37 1686 581 1992 1759 1897 855 798 1262 1738 240 1328 565 1979 308 756 416 1953 817 1204 890 1340 1495 261 1201 1100 111 1338 864 847 881 1986 847 998 92 882 398 1700 1992 581 727 1328 156 1472 1026 989 1523 1264 1636 581 1237 875 1081 945 1093 282 660 ...
result:
ok Accepted!
Test #3:
score: 10
Accepted
time: 684ms
memory: 50232kb
input:
200000 56968 132021 105656 107536 123627 58349 119191 138198 133708 142638 114350 24980 137784 40346 124158 130204 80684 183207 78156 94449 21893 157560 54464 73571 145830 57756 160288 32120 178632 142663 26565 185985 70576 24906 32217 115826 185756 137673 54280 179613 77826 144447 66005 29955 11745...
output:
174691 1521 23174
result:
ok Accepted!
Test #4:
score: 10
Accepted
time: 693ms
memory: 50312kb
input:
200000 41999 100683 85781 129266 122431 179332 162416 44814 24405 42267 154161 12483 178049 159964 67625 152391 133072 25866 178005 14695 94384 170290 54701 40323 66280 128515 159022 55057 14985 12920 182805 40659 173117 67973 99771 26102 198919 94543 23608 187601 61125 5759 89437 47647 18142 192402...
output:
65084 113267 192812
result:
ok Accepted!
Test #5:
score: 10
Accepted
time: 713ms
memory: 50368kb
input:
200000 195072 75458 31991 127114 60943 49502 186375 1130 45394 147217 168455 84307 132752 188952 101108 130004 107490 22003 16275 187645 111002 42669 138880 137115 112688 172751 81697 99037 166996 18495 2234 56119 170807 101349 105736 23180 148159 40863 136678 11849 190707 91245 61779 120740 157115 ...
output:
189386 1424 70423 1975 195394 153762 90968 80320 7013 193713 128042 128393 129871 183866 8567 86400 155516 67359 148566 119536 75836 117800 147523 19695 57574 64251 185737 91061 117656 56887
result:
ok Accepted!
Test #6:
score: 10
Accepted
time: 712ms
memory: 50168kb
input:
200000 48556 78408 155026 9376 8983 61211 150393 85068 90892 109283 75746 89742 6760 187245 168658 130735 68602 127646 60802 149828 22898 59471 172845 100274 42303 190696 7612 134905 94702 59800 129633 192496 19903 64869 51318 63358 34692 66030 98535 176606 153647 177529 157903 147292 106273 107278 ...
output:
181745 194891 117235 105797 173570 81522 167539 178085 105571 64702 3606 62620 100291 191330 186924 7628 11383 134440 20026 19728 91691 49868 110284 99352 46632 164007 62199 148544 104901 195732
result:
ok Accepted!
Test #7:
score: 10
Accepted
time: 975ms
memory: 85128kb
input:
200000 1 2 2 3 3 4 4 5 5 6 6 7 7 8 8 9 9 10 10 11 11 12 12 13 13 14 14 15 15 16 16 17 17 18 18 19 19 20 20 21 21 22 22 23 23 24 24 25 25 26 26 27 27 28 28 29 29 30 30 31 31 32 32 33 33 34 34 35 35 36 36 37 37 38 38 39 39 40 40 41 41 42 42 43 43 44 44 45 45 46 46 47 47 48 48 49 49 50 50 51 51 52 52 5...
output:
167533 21889 84837 55034 84837 123793 179616 64133 84837 84837 104948 68167 120812 79182 84837 171322 84837 56707 84837 28666 143450 7209 84837 179077 174290 1 131964 24802 45765 199989 57460 60359 200000 145198 84837 63126 80810 101392 84837 178120 84837 882 175870 76688 84837 166973 84837 71995 63...
result:
ok Accepted!
Test #8:
score: 0
Time Limit Exceeded
input:
200000 5732 55198 128966 25317 114737 116391 150003 18274 43867 70745 76222 4169 55976 114951 198396 72896 38647 19711 12756 172119 73197 117994 117512 14177 130965 126990 119440 183341 142023 60829 111893 57350 122754 123305 36525 79077 36447 91967 135405 170456 165839 147481 66074 175822 22238 264...
output:
result:
Test #9:
score: 0
Time Limit Exceeded
input:
200000 185063 17064 180987 114492 88071 71803 158363 135918 60910 54848 97338 6734 192937 9410 49617 199068 82499 63554 188791 188152 178767 40866 11304 27212 144234 138097 42236 3946 103355 12683 50992 20598 145723 48620 11709 115688 123172 121379 70541 130844 147827 39431 139372 61280 42705 54015 ...
output:
result:
Test #10:
score: 0
Time Limit Exceeded
input:
200000 197244 185999 18639 124754 154223 12099 53676 167616 22710 22294 150412 66132 19320 75478 170410 122661 130961 175554 171586 85572 188386 81238 120249 117687 43214 126266 8744 165654 164725 189519 124144 170329 86605 100197 130545 17030 113301 96665 67733 187286 37846 146399 75352 117550 3235...