#pragma GCC optimize("O3,unroll-loops")
#pragma GCC targent("avx,popcnt,sse4,abm")
#include <bits/stdc++.h>
using namespace std;
#define int long long
struct SGTR{
int l;int r;
int lz;
int mn;
}tr[800055];
void buildtr(int nw,int l,int r){
tr[nw].l=l;tr[nw].r=r;
tr[nw].lz=0;
tr[nw].mn=0;
if(l==r){return;}
int mi=(l+r)/2;
buildtr(nw*2,l,mi);buildtr(nw*2+1,mi+1,r);
}
void psh(int nw){
int v=tr[nw].lz;
tr[nw].lz=0;
tr[nw*2].lz+=v;tr[nw*2+1].lz+=v;
tr[nw*2].mn+=v;tr[nw*2+1].mn+=v;
}
int qmn(int nw,int l,int r){
if(r<l){return 1e9;}
if(tr[nw].l>=l&&tr[nw].r<=r){
return tr[nw].mn;
}
if(tr[nw].l>r||tr[nw].r<l){return 1e9;}
psh(nw);
return min(qmn(nw*2,l,r),qmn(nw*2+1,l,r));
}
void upd(int nw,int l,int r,int v){
if(r<l){return;}
if(tr[nw].l>=l&&tr[nw].r<=r){
tr[nw].mn+=v;tr[nw].lz+=v;
return;
}
if(tr[nw].l>r||tr[nw].r<l){return;}
psh(nw);
upd(nw*2,l,r,v);upd(nw*2+1,l,r,v);
tr[nw].mn=min(tr[nw*2].mn,tr[nw*2+1].mn);
}
int n;int M;
vector<int>e[500005];
int D;
int rt1;int rt2;
int dph[500005];
int dp[500005];
int fa[500005];
void dpdfs(int nw,int pa){
fa[nw]=pa;
dph[nw]=dph[pa]+1;
dp[nw]=nw;
for(int i=0;i<e[nw].size();i++){
int v=e[nw][i];
if(v==pa){continue;}
dpdfs(v,nw);
if(dph[dp[v]]>dph[dp[nw]]){
dp[nw]=dp[v];
}
}
}
void finrt(){
dpdfs(1,0);
rt1=dp[1];
//cout<<rt1<<" ";
dpdfs(rt1,0);
rt2=dp[rt1];
//cout<<rt2<<endl;
D=dph[rt2];
}
void srtlng(int rt){
dpdfs(rt,0);
for(int nw=1;nw<=n;nw++){
for(int i=0;i<e[nw].size();i++){
int v=e[nw][i];
if(v==fa[nw]){continue;}
if(e[nw][0]==fa[nw]||dph[dp[v]]>dph[dp[e[nw][0]]]){
swap(e[nw][i],e[nw][0]);
}
}
for(int i=1;i<e[nw].size();i++){
int v=e[nw][i];
if(v==fa[nw]){continue;}
if(e[nw][1]==fa[nw]||dph[dp[v]]>dph[dp[e[nw][1]]]){
swap(e[nw][i],e[nw][1]);
}
}
}
}
int ar[500005];
int buc[500005];
int bcnt;
void add(int v){
buc[v]++;
if(buc[v]==1){bcnt++;}
//cout<<v<<" "<<bcnt<<endl;
}
void del(int v){
buc[v]--;
if(buc[v]==0){bcnt--;}
// cout<<v<<" "<<bcnt<<endl;
}
stack<int>stk;
int ans[500005];
void dfs(int nw,int pa){
if(e[nw].size()==1&&pa){
ans[nw]=bcnt;
return;
}
vector<int>rev;
if(1){
int d=0;
if(e[nw].size()>=2&&e[nw][1]!=pa){
d=dph[dp[e[nw][1]]]-dph[nw];
}
while(stk.size()&&dph[stk.top()]>=dph[nw]-d){
rev.push_back(stk.top());
del(ar[stk.top()]);
stk.pop();
}
stk.push(nw);
add(ar[nw]);
dfs(e[nw][0],nw);
del(ar[nw]);
stk.pop();
}
int d=dph[dp[e[nw][0]]]-dph[nw];
while(stk.size()&&dph[stk.top()]>=dph[nw]-d){
rev.push_back(stk.top());
del(ar[stk.top()]);
stk.pop();
}
ans[nw]=max(ans[nw],bcnt);
add(ar[nw]);
stk.push(nw);
for(int i=1;i<e[nw].size();i++){
int v=e[nw][i];
if(v==pa){continue;}
dfs(v,nw);
}
del(ar[nw]);
stk.pop();
while(rev.size()){
add(ar[rev.back()]);
stk.push(rev.back());
rev.pop_back();
}
}
void solve(int rt){
bcnt=0;
while(stk.size()){stk.pop();}
for(int i=1;i<=n;i++){
buc[i]=0;
}
dpdfs(rt,0);
srtlng(rt);
//buildtr(1,1,n);
dfs(rt,0);
}
signed main(){
cin.tie(0);
ios_base::sync_with_stdio(0);
cin>>n>>M;
for(int i=1;i<=n-1;i++){
int a;int b;
cin>>a>>b;
e[a].push_back(b);
e[b].push_back(a);
}
for(int i=1;i<=n;i++){cin>>ar[i];}
finrt();
//cout<<rt1<<endl;
solve(rt1);
//cout<<endl;
//cout<<rt2<<endl;
solve(rt2);
//cout<<rt1<<" "<<rt2<<endl;
for(int i=1;i<=n;i++){
cout<<ans[i]<<endl;
}
}
Compilation message
joi2019_ho_t5.cpp:2: warning: ignoring '#pragma GCC targent' [-Wunknown-pragmas]
2 | #pragma GCC targent("avx,popcnt,sse4,abm")
|
joi2019_ho_t5.cpp: In function 'void dpdfs(long long int, long long int)':
joi2019_ho_t5.cpp:57:14: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
57 | for(int i=0;i<e[nw].size();i++){
| ~^~~~~~~~~~~~~
joi2019_ho_t5.cpp: In function 'void srtlng(long long int)':
joi2019_ho_t5.cpp:80:18: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
80 | for(int i=0;i<e[nw].size();i++){
| ~^~~~~~~~~~~~~
joi2019_ho_t5.cpp:87:18: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
87 | for(int i=1;i<e[nw].size();i++){
| ~^~~~~~~~~~~~~
joi2019_ho_t5.cpp: In function 'void dfs(long long int, long long int)':
joi2019_ho_t5.cpp:144:14: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
144 | for(int i=1;i<e[nw].size();i++){
| ~^~~~~~~~~~~~~
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
7 ms |
12116 KB |
Output is correct |
| 2 |
Incorrect |
10 ms |
12232 KB |
Output isn't correct |
| 3 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
215 ms |
22224 KB |
Output is correct |
| 2 |
Execution timed out |
2092 ms |
35100 KB |
Time limit exceeded |
| 3 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
345 ms |
25824 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
7 ms |
12116 KB |
Output is correct |
| 2 |
Incorrect |
10 ms |
12232 KB |
Output isn't correct |
| 3 |
Halted |
0 ms |
0 KB |
- |
