Submission #1734610
Source Code Expand
#include<cstdio> #include<cstring> #include<iomanip> #include<iostream> #include<algorithm> #include<cmath> #include<ctime> #include<map> #include<set> #include<vector> #include<queue> #include<bitset> #include<stack> #include<cassert> #define F first #define S second #define mp make_pair #define pii pair<int,int> #define pb push_back #define mem(x,y) memset(x,y,sizeof x) #define sqr(x) ((x)*(x)) using namespace std; typedef long long ll; typedef double db; const int INF=2e9,N=100010; const db eps=1e-8; inline void read(int &x) { x=0; int f=1; char ch=getchar(); while( (ch<'0' || ch>'9') && ch!='-') ch=getchar(); if(ch=='-') {f=-1; ch=getchar();} while(ch>='0' && ch <='9') x=x*10+ch-'0',ch=getchar(); x*=f; } inline void read(ll &x) { x=0; int f=1; char ch=getchar(); while( (ch<'0' || ch>'9') && ch!='-') ch=getchar(); if(ch=='-') {f=-1; ch=getchar();} while(ch>='0' && ch <='9') x=x*10+ch-'0',ch=getchar(); x*=f; } //==========================head template========================== int n,a[N]; int fa[N]; bool ans[N]; int head[N],to[N<<1],nxt[N<<1],lst=0; inline void adde(int x,int y) {nxt[++lst]=head[x]; to[lst]=y; head[x]=lst;} inline bool ok(int x,int p) {x=x%((int)pow(10,p+1)); return x/(int)pow(10,p)!=0;} inline void dfs1(int u) { for(int i=head[u];i;i=nxt[i]) if(to[i]!=fa[u]) { fa[to[i]]=u; dfs1(to[i]); if(!ans[to[i]] && a[u]>a[to[i]]) ans[u]=1; } } void dfs2(int u) { for(int i=head[u];i;i=nxt[i]) if(to[i]!=fa[u]) { if(!ans[u] && a[to[i]]>a[u]) ans[to[i]]=1; dfs2(to[i]); } } int main() { // freopen("C.in","r",stdin); // freopen("C.out","w",stdout); read(n);int Sta=0; for(int i=1;i<=n;i++) read(a[i]); for(int i=1;i<n;i++) { int x,y; read(x); read(y); adde(x,y); adde(y,x); } dfs1(1); // for(int i=1;i<=n;i++) printf("%d ",ans[i]); dfs2(1); for(int i=1;i<=n;i++) if(ans[i]) printf("%d ",i); return 0; } /*5 5 4 1 2 3 1 2 1 3 2 4 2 5*/
Submission Info
Submission Time | |
---|---|
Task | F - Tree Game |
User | Stevexx |
Language | C++14 (GCC 5.4.1) |
Score | 1600 |
Code Size | 2006 Byte |
Status | AC |
Exec Time | 2 ms |
Memory | 384 KB |
Judge Result
Set Name | Sample | All | ||||
---|---|---|---|---|---|---|
Score / Max Score | 0 / 0 | 1600 / 1600 | ||||
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, in48.txt, in49.txt, in5.txt, in50.txt, in6.txt, in7.txt, in8.txt, in9.txt, sample1.txt, sample2.txt, sample3.txt |
Case Name | Status | Exec Time | Memory |
---|---|---|---|
in1.txt | AC | 2 ms | 384 KB |
in10.txt | AC | 2 ms | 384 KB |
in11.txt | AC | 2 ms | 384 KB |
in12.txt | AC | 1 ms | 256 KB |
in13.txt | AC | 1 ms | 256 KB |
in14.txt | AC | 2 ms | 384 KB |
in15.txt | AC | 2 ms | 384 KB |
in16.txt | AC | 2 ms | 384 KB |
in17.txt | AC | 2 ms | 384 KB |
in18.txt | AC | 2 ms | 384 KB |
in19.txt | AC | 2 ms | 384 KB |
in2.txt | AC | 2 ms | 384 KB |
in20.txt | AC | 2 ms | 384 KB |
in21.txt | AC | 2 ms | 384 KB |
in22.txt | AC | 2 ms | 384 KB |
in23.txt | AC | 2 ms | 384 KB |
in24.txt | AC | 2 ms | 384 KB |
in25.txt | AC | 2 ms | 384 KB |
in26.txt | AC | 2 ms | 384 KB |
in27.txt | AC | 2 ms | 384 KB |
in28.txt | AC | 2 ms | 384 KB |
in29.txt | AC | 2 ms | 384 KB |
in3.txt | AC | 2 ms | 384 KB |
in30.txt | AC | 2 ms | 384 KB |
in31.txt | AC | 2 ms | 384 KB |
in32.txt | AC | 2 ms | 384 KB |
in33.txt | AC | 2 ms | 384 KB |
in34.txt | AC | 2 ms | 384 KB |
in35.txt | AC | 2 ms | 384 KB |
in36.txt | AC | 1 ms | 256 KB |
in37.txt | AC | 1 ms | 256 KB |
in38.txt | AC | 1 ms | 256 KB |
in39.txt | AC | 1 ms | 256 KB |
in4.txt | AC | 2 ms | 384 KB |
in40.txt | AC | 2 ms | 384 KB |
in41.txt | AC | 2 ms | 384 KB |
in42.txt | AC | 2 ms | 384 KB |
in43.txt | AC | 2 ms | 384 KB |
in44.txt | AC | 2 ms | 384 KB |
in45.txt | AC | 2 ms | 384 KB |
in46.txt | AC | 2 ms | 384 KB |
in47.txt | AC | 2 ms | 384 KB |
in48.txt | AC | 2 ms | 384 KB |
in49.txt | AC | 2 ms | 384 KB |
in5.txt | AC | 2 ms | 384 KB |
in50.txt | AC | 2 ms | 384 KB |
in6.txt | AC | 2 ms | 384 KB |
in7.txt | AC | 2 ms | 384 KB |
in8.txt | AC | 2 ms | 384 KB |
in9.txt | AC | 2 ms | 384 KB |
sample1.txt | AC | 1 ms | 256 KB |
sample2.txt | AC | 1 ms | 256 KB |
sample3.txt | AC | 1 ms | 256 KB |