結果

問題 No.2634 Tree Distance 3
ユーザー maksimmaksim
提出日時 2024-02-16 22:13:43
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 974 ms / 3,000 ms
コード長 2,368 bytes
コンパイル時間 1,961 ms
コンパイル使用メモリ 178,564 KB
実行使用メモリ 86,384 KB
最終ジャッジ日時 2024-09-28 20:51:27
合計ジャッジ時間 39,416 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 579 ms
77,636 KB
testcase_01 AC 571 ms
79,048 KB
testcase_02 AC 545 ms
76,144 KB
testcase_03 AC 553 ms
78,880 KB
testcase_04 AC 562 ms
78,992 KB
testcase_05 AC 583 ms
78,668 KB
testcase_06 AC 602 ms
78,576 KB
testcase_07 AC 581 ms
78,688 KB
testcase_08 AC 498 ms
78,516 KB
testcase_09 AC 502 ms
77,044 KB
testcase_10 AC 478 ms
78,256 KB
testcase_11 AC 463 ms
78,624 KB
testcase_12 AC 486 ms
78,312 KB
testcase_13 AC 499 ms
77,868 KB
testcase_14 AC 501 ms
77,420 KB
testcase_15 AC 516 ms
78,936 KB
testcase_16 AC 713 ms
75,320 KB
testcase_17 AC 373 ms
57,584 KB
testcase_18 AC 543 ms
63,492 KB
testcase_19 AC 603 ms
69,420 KB
testcase_20 AC 163 ms
47,160 KB
testcase_21 AC 932 ms
82,836 KB
testcase_22 AC 974 ms
82,672 KB
testcase_23 AC 953 ms
86,384 KB
testcase_24 AC 938 ms
79,484 KB
testcase_25 AC 934 ms
77,356 KB
testcase_26 AC 933 ms
79,828 KB
testcase_27 AC 931 ms
78,020 KB
testcase_28 AC 942 ms
77,112 KB
testcase_29 AC 880 ms
78,528 KB
testcase_30 AC 669 ms
78,084 KB
testcase_31 AC 669 ms
76,840 KB
testcase_32 AC 655 ms
77,968 KB
testcase_33 AC 371 ms
62,620 KB
testcase_34 AC 78 ms
37,576 KB
testcase_35 AC 209 ms
51,208 KB
testcase_36 AC 129 ms
44,972 KB
testcase_37 AC 258 ms
57,856 KB
testcase_38 AC 15 ms
31,668 KB
testcase_39 AC 15 ms
31,868 KB
testcase_40 AC 14 ms
31,296 KB
testcase_41 AC 14 ms
31,556 KB
testcase_42 AC 13 ms
30,356 KB
testcase_43 AC 190 ms
45,780 KB
testcase_44 AC 123 ms
42,524 KB
testcase_45 AC 561 ms
68,452 KB
testcase_46 AC 330 ms
56,924 KB
testcase_47 AC 579 ms
75,304 KB
testcase_48 AC 608 ms
75,076 KB
testcase_49 AC 593 ms
77,412 KB
testcase_50 AC 656 ms
76,344 KB
testcase_51 AC 628 ms
77,424 KB
testcase_52 AC 614 ms
77,352 KB
testcase_53 AC 13 ms
30,400 KB
testcase_54 AC 12 ms
30,148 KB
testcase_55 AC 12 ms
30,204 KB
testcase_56 AC 12 ms
30,860 KB
testcase_57 AC 14 ms
31,736 KB
testcase_58 AC 12 ms
31,444 KB
testcase_59 AC 10 ms
31,336 KB
testcase_60 AC 551 ms
77,128 KB
testcase_61 AC 555 ms
77,280 KB
testcase_62 AC 576 ms
77,764 KB
testcase_63 AC 441 ms
76,228 KB
testcase_64 AC 281 ms
60,172 KB
testcase_65 AC 407 ms
72,388 KB
testcase_66 AC 108 ms
45,224 KB
testcase_67 AC 96 ms
40,568 KB
testcase_68 AC 448 ms
78,296 KB
testcase_69 AC 458 ms
76,836 KB
testcase_70 AC 467 ms
76,856 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

using namespace std;
#define int long long
#define app push_back
#define all(x) (x).begin(),(x).end()
#ifdef LOCAL
#define debug(...) [](auto...a){ ((cout << a << ' '), ...) << endl;}(#__VA_ARGS__, ":", __VA_ARGS__)
#else
#define debug(...)
#endif
#ifdef LOCAL
#define __int128 long long
#endif // LOCAL
const int maxn=1e6+5;
vector<int> a[maxn];
int c[maxn];
bool used[maxn];
int corn[maxn];
int u[maxn][20];
int res[maxn];
pair<int,int> cur={-1,-1};
void dfs(int x)
{
    used[x]=true;
    for(int i=1;i<20;++i) {u[x][i]=u[u[x][i-1]][i-1];}
    for(int v:a[x])
    {
        if(!used[v])
        {
            u[v][0]=x;
            corn[v]=corn[x]+1;
            dfs(v);
        }
    }
    used[x]=false;
}
int lca(int x,int y)
{
    for(int i=19;i>=0;--i) {if(corn[x]>=corn[y]+(1<<i)) x=u[x][i];}
    for(int i=19;i>=0;--i) {if(corn[y]>=corn[x]+(1<<i)) y=u[y][i];}
    for(int i=19;i>=0;--i) {if(u[x][i]!=u[y][i]) {x=u[x][i];y=u[y][i];}}
    if(x!=y) {x=u[x][0];y=u[y][0];}
    return x;
}
int dist(int x,int y)
{
    if(x==(-1) || y==(-1)) return 0;
    return corn[x]+corn[y]-2*corn[lca(x,y)];
}
void add(int pos)
{
    if(cur.first==(-1)) {cur.first=pos;return;}
    if(cur.second==(-1)) {cur.second=pos;return;}
    int was=dist(cur.first,cur.second);int x1=dist(cur.first,pos);int x2=dist(cur.second,pos);
    int ma=max({was,x1,x2});
    if(was==ma) return;
    if(x1==ma) {cur={cur.first,pos};return;}
    cur={pos,cur.second};
}
int get(int pos)
{
    return max(dist(pos,cur.first),dist(pos,cur.second));
}
int32_t main()
{
    ios_base::sync_with_stdio(false);cin.tie(0);cout.tie(0);
    int n;cin>>n;
    for(int i=0;i<n;++i) cin>>c[i];
    for(int i=0;i<n-1;++i) {int x,y;cin>>x>>y;--x;--y;a[x].app(y);a[y].app(x);}
    vector<int> v(n);iota(all(v),0LL);sort(all(v),[&](int i,int j) {return make_pair(c[i],i)<make_pair(c[j],j);});
    dfs(0);
    for(int i=n-1;i>=0;--i)
    {
        int pos=v[i];
        if(i==n-1) {add(pos);continue;}
        res[pos]=max(res[pos],get(pos));
        add(pos);
    }
    cur={-1,-1};
    sort(all(v),[&](int i,int j) {return make_pair(c[i],-i)<make_pair(c[j],-j);});
    for(int i=n-1;i>=0;--i)
    {
        int pos=v[i];
        if(i==n-1) {add(pos);continue;}
        res[pos]=max(res[pos],get(pos));
        add(pos);
    }
    for(int i=0;i<n;++i) cout<<res[i]<<' ';
    return 0;
}
0