結果

問題 No.828 全方位神童数
ユーザー koprickykopricky
提出日時 2019-05-04 04:21:51
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 195 ms / 2,000 ms
コード長 3,552 bytes
コンパイル時間 1,728 ms
コンパイル使用メモリ 173,244 KB
実行使用メモリ 28,156 KB
最終ジャッジ日時 2024-06-11 04:21:53
合計ジャッジ時間 9,095 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 4 ms
8,064 KB
testcase_01 AC 4 ms
8,320 KB
testcase_02 AC 3 ms
8,192 KB
testcase_03 AC 10 ms
8,832 KB
testcase_04 AC 6 ms
8,320 KB
testcase_05 AC 9 ms
8,832 KB
testcase_06 AC 6 ms
8,192 KB
testcase_07 AC 5 ms
8,192 KB
testcase_08 AC 5 ms
8,192 KB
testcase_09 AC 5 ms
8,192 KB
testcase_10 AC 8 ms
8,576 KB
testcase_11 AC 5 ms
8,192 KB
testcase_12 AC 8 ms
8,704 KB
testcase_13 AC 54 ms
15,676 KB
testcase_14 AC 128 ms
23,888 KB
testcase_15 AC 42 ms
14,464 KB
testcase_16 AC 61 ms
16,300 KB
testcase_17 AC 48 ms
14,336 KB
testcase_18 AC 85 ms
18,236 KB
testcase_19 AC 162 ms
28,156 KB
testcase_20 AC 103 ms
22,376 KB
testcase_21 AC 143 ms
25,704 KB
testcase_22 AC 32 ms
12,672 KB
testcase_23 AC 12 ms
8,832 KB
testcase_24 AC 79 ms
14,032 KB
testcase_25 AC 32 ms
10,624 KB
testcase_26 AC 164 ms
19,044 KB
testcase_27 AC 138 ms
17,828 KB
testcase_28 AC 153 ms
19,064 KB
testcase_29 AC 152 ms
18,780 KB
testcase_30 AC 76 ms
14,008 KB
testcase_31 AC 70 ms
13,896 KB
testcase_32 AC 154 ms
19,004 KB
testcase_33 AC 188 ms
19,800 KB
testcase_34 AC 145 ms
17,764 KB
testcase_35 AC 83 ms
14,264 KB
testcase_36 AC 107 ms
15,420 KB
testcase_37 AC 66 ms
13,464 KB
testcase_38 AC 100 ms
15,176 KB
testcase_39 AC 170 ms
19,060 KB
testcase_40 AC 66 ms
13,448 KB
testcase_41 AC 55 ms
12,724 KB
testcase_42 AC 195 ms
19,724 KB
testcase_43 AC 102 ms
28,044 KB
testcase_44 AC 41 ms
15,616 KB
testcase_45 AC 61 ms
19,592 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#define ll long long
#define INF 1000000005
#define MOD 1000000007
#define EPS 1e-10
#define rep(i,n) for(int i=0;i<(int)(n);++i)
#define rrep(i,n) for(int i=(int)(n)-1;i>=0;--i)
#define srep(i,s,t) for(int i=(int)(s);i<(int)(t);++i)
#define each(a,b) for(auto& (a): (b))
#define all(v) (v).begin(),(v).end()
#define len(v) (int)(v).size()
#define zip(v) sort(all(v)),v.erase(unique(all(v)),v.end())
#define cmx(x,y) x=max(x,y)
#define cmn(x,y) x=min(x,y)
#define fi first
#define se second
#define pb push_back
#define show(x) cout<<#x<<" = "<<(x)<<endl
#define sar(a,n) cout<<#a<<":";rep(pachico,n)cout<<" "<<a[pachico];cout<<endl

using namespace std;

template<typename S,typename T>auto&operator<<(ostream&o,pair<S,T>p){return o<<"{"<<p.fi<<","<<p.se<<"}";}
template<typename T>auto&operator<<(ostream&o,set<T>s){for(auto&e:s)o<<e<<" ";return o;}
template<typename S,typename T,typename U>
auto&operator<<(ostream&o,priority_queue<S,T,U>q){while(!q.empty())o<<q.top()<<" ",q.pop();return o;}
template<typename K,typename T>auto&operator<<(ostream&o,map<K,T>m){for(auto&e:m)o<<e<<" ";return o;}
template<typename T>auto&operator<<(ostream&o,vector<T>v){for(auto&e:v)o<<e<<" ";return o;}
void ashow(){cout<<endl;}template<typename T,typename...A>void ashow(T t,A...a){cout<<t<<" ";ashow(a...);}

typedef pair<int,int> P;
typedef pair<ll,ll> pll;
typedef vector<int> vi;
typedef vector<ll> vl;
typedef vector<vi> vvi;
typedef vector<vl> vvl;
typedef vector<P> vp;
typedef vector<double> vd;
typedef vector<string> vs;

const int MAX_N = 200005;

template<typename T> class segtree {
private:
    int n,sz;
    vector<T> node;
public:
    segtree(vector<T>& v){
        sz = (int)v.size();
        n = 1;
        while(n < sz){
            n *= 2;
        }
        node.assign(2*n,numeric_limits<T>::min());
        rep(i,sz){
            node[i+n] = v[i];
        }
        for(int i=n-1; i>=1; i--){
            node[i] = max(node[2*i],node[2*i+1]);
        }
    }
    void update(int k,T a)
    {
    	node[k+=n] = a;
    	while(k>>=1){
    		node[k] = max(node[2*k],node[2*k+1]);
    	}
    }
    T query(int a,int b)
    {
        T res1 = numeric_limits<T>::min();
        T res2 = numeric_limits<T>::min();
        a += n, b += n;
        while(a != b){
            if(a % 2) res1 = max(res1, node[a++]);
            if(b % 2) res2 = max(res2, node[--b]);
            a >>= 1, b>>= 1;
        }
        return max(res1, res2);
    }
    void print()
    {
        rep(i,sz){
            cout << query(i,i+1) << " ";
        }
        cout << endl;
    }
};

vector<int> G[MAX_N], ord;
int r[MAX_N], in[MAX_N], out[MAX_N], ans[MAX_N];

void dfs(int u, int p)
{
    in[u] = len(ord);
    ord.pb(u);
    each(v, G[u]){
        if(v != p) dfs(v, u);
    }
    out[u] = len(ord);
}

void rec(int u, int p, int val, segtree<int>& sg){
    for(;;val++){
        int ver = sg.query(in[u], out[u]);
        if(ver == -INF) break;
        ans[ver] = val + 1;
        each(nx, G[ver]){
            if(in[nx] > in[ver]) rec(nx, ver, val + 1, sg);
        }
        sg.update(in[ver], -INF);
    }
}

int main()
{
    cin.tie(0);
    ios::sync_with_stdio(false);
    int n;
    cin >> n;
    rep(i,n){
        cin >> r[i];
    }
    rep(i,n-1){
        int u, v;
        cin >> u >> v;
        G[u-1].pb(v-1), G[v-1].pb(u-1);
    }
    dfs(0, -1);
    segtree<int> sg(ord);
    rec(0, -1, 0, sg);
    int score = 1;
    rep(i,n){
        score = (ll)score * (r[i] + ans[i]) % MOD;
    }
    cout << score << "\n";
    return 0;
}
0