Submission #1732046
Source Code Expand
#include <bits/stdc++.h> #include <ext/pb_ds/assoc_container.hpp> #include <ext/pb_ds/tree_policy.hpp> using namespace std; using namespace __gnu_pbds; #define fi first #define se second #define mp make_pair #define pb push_back #define fbo find_by_order #define ook order_of_key typedef long long ll; typedef pair<int,int> ii; typedef vector<int> vi; typedef long double ld; typedef tree<int, null_type, less<int>, rb_tree_tag, tree_order_statistics_node_update> pbds; typedef set<int>::iterator sit; typedef map<int,int>::iterator mit; typedef vector<int>::iterator vit; int par[111111]; set<ii> R,B; vi adjr[111111]; vi adjb[111111]; int rt(int u) { if(par[u]<0) return u; else return (par[u]=rt(par[u])); } queue<ii> q; int res; void merge(int u, int v) { u=rt(u); v=rt(v); if(u==v) return ; res++; if(par[u]>par[v]) swap(u,v); //v merges to u for(int i=0;i<adjr[v].size();i++) { int val=adjr[v][i]; if(val==u) continue; adjr[u].pb(val); R.insert(mp(u,val)); if(B.find(mp(u,val))!=B.end()||B.find(mp(val,u))!=B.end()) q.push(mp(u,val)); } for(int i=0;i<adjb[v].size();i++) { int val=adjb[v][i]; if(val==u) continue; adjb[u].pb(val); B.insert(mp(u,val)); if(R.find(mp(u,val))!=R.end()||R.find(mp(val,u))!=R.end()) q.push(mp(u,val)); } par[u]+=par[v]; par[v]=u; } int main() { ios_base::sync_with_stdio(0); cin.tie(0); memset(par,-1,sizeof(par)); int n; cin>>n; for(int i=0;i<n-1;i++) { int u,v; cin>>u>>v; u--; v--; R.insert(mp(u,v)); adjr[u].pb(v); adjr[v].pb(u); } for(int i=0;i<n-1;i++) { int u,v; cin>>u>>v; u--; v--; B.insert(mp(u,v)); adjb[u].pb(v); adjb[v].pb(u); if(R.find(mp(u,v))!=R.end()||R.find(mp(v,u))!=R.end()) { q.push(mp(u,v)); } } res=0; while(!q.empty()) { int u=q.front().fi; int v=q.front().se; q.pop(); merge(u,v); } cout<<(res==n-1?"YES":"NO")<<'\n'; }
Submission Info
Submission Time | |
---|---|
Task | E - Blue and Red Tree |
User | vjudge3 |
Language | C++14 (GCC 5.4.1) |
Score | 0 |
Code Size | 1869 Byte |
Status | WA |
Exec Time | 692 ms |
Memory | 32748 KB |
Judge Result
Set Name | Sample | All | ||||||
---|---|---|---|---|---|---|---|---|
Score / Max Score | 0 / 0 | 0 / 1400 | ||||||
Status |
|
|
Set Name | Test Cases |
---|---|
Sample | sample1.txt, sample2.txt, sample3.txt |
All | sample1.txt, sample2.txt, sample3.txt, in1.txt, in10.txt, in11.txt, in12.txt, in13.txt, in14.txt, in15.txt, in16.txt, in17.txt, in18.txt, in19.txt, in2.txt, in20.txt, in21.txt, in22.txt, in23.txt, in24.txt, in25.txt, in26.txt, in27.txt, in28.txt, in29.txt, in3.txt, in30.txt, in31.txt, in32.txt, in33.txt, in34.txt, in35.txt, in36.txt, in37.txt, in38.txt, in39.txt, in4.txt, in40.txt, in41.txt, in42.txt, in43.txt, in44.txt, in45.txt, in46.txt, in47.txt, in5.txt, in6.txt, in7.txt, in8.txt, in9.txt, sample1.txt, sample2.txt, sample3.txt |
Case Name | Status | Exec Time | Memory |
---|---|---|---|
in1.txt | AC | 5 ms | 6012 KB |
in10.txt | AC | 217 ms | 21760 KB |
in11.txt | WA | 407 ms | 30208 KB |
in12.txt | WA | 412 ms | 30336 KB |
in13.txt | WA | 407 ms | 30208 KB |
in14.txt | WA | 430 ms | 30336 KB |
in15.txt | WA | 422 ms | 30336 KB |
in16.txt | WA | 444 ms | 30208 KB |
in17.txt | WA | 404 ms | 30336 KB |
in18.txt | WA | 423 ms | 30208 KB |
in19.txt | WA | 433 ms | 30208 KB |
in2.txt | AC | 4 ms | 5888 KB |
in20.txt | WA | 411 ms | 30208 KB |
in21.txt | AC | 445 ms | 30848 KB |
in22.txt | AC | 431 ms | 30848 KB |
in23.txt | AC | 423 ms | 30848 KB |
in24.txt | AC | 417 ms | 30720 KB |
in25.txt | AC | 430 ms | 30720 KB |
in26.txt | AC | 429 ms | 31360 KB |
in27.txt | WA | 445 ms | 31488 KB |
in28.txt | WA | 456 ms | 31360 KB |
in29.txt | WA | 449 ms | 31488 KB |
in3.txt | AC | 4 ms | 5888 KB |
in30.txt | WA | 447 ms | 31488 KB |
in31.txt | AC | 572 ms | 32748 KB |
in32.txt | AC | 571 ms | 32748 KB |
in33.txt | AC | 580 ms | 32748 KB |
in34.txt | AC | 613 ms | 32748 KB |
in35.txt | AC | 692 ms | 32748 KB |
in36.txt | AC | 614 ms | 32748 KB |
in37.txt | AC | 666 ms | 32748 KB |
in38.txt | AC | 643 ms | 32620 KB |
in39.txt | AC | 599 ms | 32620 KB |
in4.txt | AC | 226 ms | 21760 KB |
in40.txt | AC | 582 ms | 32620 KB |
in41.txt | AC | 568 ms | 32748 KB |
in42.txt | AC | 581 ms | 32748 KB |
in43.txt | AC | 574 ms | 32748 KB |
in44.txt | AC | 588 ms | 32748 KB |
in45.txt | AC | 565 ms | 32748 KB |
in46.txt | AC | 554 ms | 32620 KB |
in47.txt | AC | 558 ms | 32620 KB |
in5.txt | AC | 217 ms | 21760 KB |
in6.txt | AC | 561 ms | 32620 KB |
in7.txt | AC | 220 ms | 21760 KB |
in8.txt | AC | 219 ms | 21760 KB |
in9.txt | AC | 220 ms | 21760 KB |
sample1.txt | AC | 4 ms | 5888 KB |
sample2.txt | AC | 4 ms | 5888 KB |
sample3.txt | AC | 4 ms | 5888 KB |