結果

問題 No.828 全方位神童数
ユーザー smiken_61smiken_61
提出日時 2019-05-03 22:42:28
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 197 ms / 2,000 ms
コード長 9,359 bytes
コンパイル時間 2,333 ms
コンパイル使用メモリ 155,108 KB
実行使用メモリ 39,088 KB
最終ジャッジ日時 2023-08-30 06:30:49
合計ジャッジ時間 8,707 ms
ジャッジサーバーID
(参考情報)
judge13 / judge15
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 9 ms
20,624 KB
testcase_01 AC 9 ms
20,544 KB
testcase_02 AC 10 ms
20,540 KB
testcase_03 AC 16 ms
21,124 KB
testcase_04 AC 11 ms
20,576 KB
testcase_05 AC 15 ms
21,144 KB
testcase_06 AC 11 ms
20,548 KB
testcase_07 AC 11 ms
20,572 KB
testcase_08 AC 11 ms
20,544 KB
testcase_09 AC 10 ms
20,528 KB
testcase_10 AC 14 ms
20,804 KB
testcase_11 AC 11 ms
20,628 KB
testcase_12 AC 14 ms
21,128 KB
testcase_13 AC 60 ms
24,756 KB
testcase_14 AC 131 ms
29,588 KB
testcase_15 AC 45 ms
23,692 KB
testcase_16 AC 64 ms
25,092 KB
testcase_17 AC 51 ms
24,300 KB
testcase_18 AC 94 ms
26,888 KB
testcase_19 AC 163 ms
31,664 KB
testcase_20 AC 106 ms
27,924 KB
testcase_21 AC 142 ms
30,112 KB
testcase_22 AC 36 ms
23,064 KB
testcase_23 AC 17 ms
21,388 KB
testcase_24 AC 86 ms
26,956 KB
testcase_25 AC 37 ms
23,232 KB
testcase_26 AC 182 ms
31,892 KB
testcase_27 AC 153 ms
30,404 KB
testcase_28 AC 187 ms
32,040 KB
testcase_29 AC 174 ms
31,416 KB
testcase_30 AC 88 ms
26,952 KB
testcase_31 AC 85 ms
26,648 KB
testcase_32 AC 179 ms
31,996 KB
testcase_33 AC 197 ms
33,016 KB
testcase_34 AC 148 ms
30,364 KB
testcase_35 AC 95 ms
27,180 KB
testcase_36 AC 121 ms
28,776 KB
testcase_37 AC 76 ms
26,168 KB
testcase_38 AC 109 ms
28,276 KB
testcase_39 AC 177 ms
32,196 KB
testcase_40 AC 73 ms
26,080 KB
testcase_41 AC 62 ms
25,092 KB
testcase_42 AC 196 ms
33,048 KB
testcase_43 AC 92 ms
39,088 KB
testcase_44 AC 42 ms
27,752 KB
testcase_45 AC 57 ms
31,512 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

     #include <bits/stdc++.h>

//    #include <boost/multiprecision/cpp_int.hpp>
 #define int long long
 #define inf  1000000007
 #define pa pair<int,int>
 #define ll long long
 #define pal pair<double,double>
 #define ppap pair<pa,int>
  #define PI 3.14159265358979323846
  #define paa pair<int,char>
  #define  mp make_pair
  #define  pb push_back
  #define EPS (1e-10)
                                          
    int dx[8]={0,1,0,-1,1,1,-1,-1};
    int dy[8]={1,0,-1,0,-1,1,1,-1};
                                            using namespace std;
                                   			class pa3{
                                            	public:
                                            	int x;
                                   				int y,z;
                                            	pa3(int x=0,int y=0,int z=0):x(x),y(y),z(z) {}
                                            	bool operator < (const pa3 &p) const{
                                            		if(x!=p.x) return x<p.x;
                                            		if(y!=p.y) return y<p.y;
                                            		 return z<p.z;
                                            		//return x != p.x ? x<p.x: y<p.y;
                                            	}
                                   				bool operator > (const pa3 &p) const{
                                            		if(x!=p.x) return x>p.x;
                                            		if(y!=p.y) return y>p.y;
                                            		 return z>p.z;
                                            		//return x != p.x ? x<p.x: y<p.y;
                                            	}
                                            	bool operator == (const pa3 &p) const{
                                            		return x==p.x && y==p.y && z==p.z;
                                            	}
                                            		bool operator != (const pa3 &p) const{
                                            			return !( x==p.x && y==p.y && z==p.z);
                                            	}
                                            
                                            };
                                            
                                            class pa4{
                                            	public:
                                            	int x;
                                            	int y,z,w;
                                            	pa4(int x=0,int y=0,int z=0,int w=0):x(x),y(y),z(z),w(w) {}
                                            	bool operator < (const pa4 &p) const{
                                            		if(x!=p.x) return x<p.x;
                                            		if(y!=p.y) return y<p.y;
                                            		if(z!=p.z)return z<p.z;
                                            		return w<p.w;
                                            		//return x != p.x ? x<p.x: y<p.y;
                                            	}
                                            	bool operator > (const pa4 &p) const{
                                            		if(x!=p.x) return x>p.x;
                                            		if(y!=p.y) return y>p.y;
                                            		if(z!=p.z)return z>p.z;
                                            		return w>p.w;
                                            		//return x != p.x ? x<p.x: y<p.y;
                                            	}
                                            	bool operator == (const pa4 &p) const{
                                            		return x==p.x && y==p.y && z==p.z &&w==p.w;
                                            	}
                                            		
                                            
                                            };
                                            class pa2{
                                            	public:
                                            	int x,y;
                                            	pa2(int x=0,int y=0):x(x),y(y) {}
                                            	pa2 operator + (pa2 p) {return pa2(x+p.x,y+p.y);}
                                            	pa2 operator - (pa2 p) {return pa2(x-p.x,y-p.y);}
                                            	bool operator < (const pa2 &p) const{
                                            		return y != p.y ? y<p.y: x<p.x;
                                            	}
                                            	bool operator > (const pa2 &p) const{
                                            		return x != p.x ? x<p.x: y<p.y;
                                            	}
                                            	bool operator == (const pa2 &p) const{
                                            		return abs(x-p.x)==0 && abs(y-p.y)==0;
                                            	}
                                            	bool operator != (const pa2 &p) const{
                                            		return !(abs(x-p.x)==0 && abs(y-p.y)==0);
                                            	}
                                            		
                                            
                                            };
                                            

                      
                                string itos( int i ) {
                                ostringstream s ;
                                s << i ;
                                return s.str() ;
                                }
                                 
                                int gcd(int v,int b){
                                	if(v>b) return gcd(b,v);
                                	if(v==b) return b;
                                	if(b%v==0) return v;
                                	return gcd(v,b%v);
                                }
                 
                            
                                int mod;
int extgcd(int a, int b, int &x, int &y) {
    if (b == 0) {
        x = 1;
        y = 0;
        return a;
    }
    int d = extgcd(b, a%b, y, x);
    y -= a/b * x;
    return d;
}
pa operator+(const pa & l,const pa & r) {   
    return {l.first+r.first,l.second+r.second};                                    
}    
pa operator-(const pa & l,const pa & r) {   
    return {l.first-r.first,l.second-r.second};                                    
}  
                                
                int pr[1200010];
                int inv[1200010];
                
                int beki(int wa,int rr,int warukazu){
                	if(rr==0) return 1%warukazu;
                	if(rr==1) return wa%warukazu;
                	wa%=warukazu;
                	if(rr%2==1) return ((ll)beki(wa,rr-1,warukazu)*(ll)wa)%warukazu;
                	ll zx=beki(wa,rr/2,warukazu);
                	return (zx*zx)%warukazu;
                }

                
    			int comb(int nn,int rr){
    				if(rr<0 || rr>nn || nn<0) return 0;
    				int r=pr[nn]*inv[rr];
    				r%=mod;
    				r*=inv[nn-rr];
    				r%=mod;
    				return r;
    			}
                
                void gya(int ert){
                	pr[0]=1;
                	for(int i=1;i<=ert;i++){
                		pr[i]=(pr[i-1]*i)%mod;
                	}
                		inv[ert]=beki(pr[ert],mod-2,mod);
                	for(int i=ert-1;i>=0;i--){
                		inv[i]=inv[i+1]*(i+1)%mod;
                	}
                }
                
              //   cin.tie(0);
    		//	ios::sync_with_stdio(false);
    			//priority_queue<pa3,vector<pa3>,greater<pa3>> pq;            
                 //sort(ve.begin(),ve.end(),greater<int>());
//    mt19937(clock_per_sec);
  //  mt19937_64 rng(chrono::steady_clock::now().time_since_epoch().count()) ;
      
       
       
                               //----------------kokomade tenpure------------
vector<int> G[200020];
struct unionfind{
	private:
	public:
	
vector<int> par,ranks,kosuu,dekai;
	
	void shoki(int N){
		par.resize(N+1,0);
		ranks.resize(N+1,0);
	
		dekai.resize(N+1,1);
		
		for(int i=0;i<=N;i++){
			par[i]=i,dekai[i]=i;
		}
	}
	
	int de(int x){
		return dekai[root(x)];
	}
	
	int root(int x){
		return par[x]==x ? x : par[x]=root(par[x]);
	}

	bool same(int x,int y){
		return root(x)==root(y);
	}
	bool is_root(int x){
		return x==root(x);
	}
	void unite(int x,int y){
 		x=root(x);
	 	y=root(y);
	
	 	if(x==y) return;
		if(ranks[x]<ranks[y]){
			par[x]=y;
		
			dekai[y]=max(dekai[x],dekai[y]);
		}
	 	else {
			par[y]=x;
			if(ranks[x]==ranks[y]) ranks[x]=ranks[x]+1;
	 	
	 		dekai[x]=max(dekai[y],dekai[x]);
	 	}
		return;
	}
};





int ans[200020]={};

vector<int> G2[200020];


int R[200020];

int kotae=1;
void dfs(int v,int val=1){
	
	
	for(auto ve:G2[v])dfs(ve,val+1);

	kotae*=val+R[v];
	
	kotae%=inf;
}
 signed main(){
 	

    	       cin.tie(0);
   			ios::sync_with_stdio(false);
int n;
 	cin>>n;
 	for(int i=1;i<=n;i++)cin>>R[i];
 	for(int i=0;i<n-1;i++){
 		int y,yy;
 		cin>>y>>yy;
 			G[y].pb(yy);
 		G[yy].pb(y);
 	}
 unionfind uf;
 	uf.shoki(200020);

 	
 	for(int i=1;i<=n;i++){
 
 		for(auto v:G[i])if(v<i){
 			G2[i].pb(uf.de(v));
 			uf.unite(v,i);
 		}
 		
 	}
 	
 	dfs(n);
 	
 	//for(int i=1;i<=n;i++)cout<<ans[i]<<endl;
 	
 	cout<<kotae<<endl;
  	return 0;
  }
  
0