結果
問題 | No.2214 Products on Tree |
ユーザー | 👑 potato167 |
提出日時 | 2023-02-10 22:00:56 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 232 ms / 3,000 ms |
コード長 | 3,002 bytes |
コンパイル時間 | 2,687 ms |
コンパイル使用メモリ | 214,812 KB |
実行使用メモリ | 28,136 KB |
最終ジャッジ日時 | 2023-09-21 23:15:40 |
合計ジャッジ時間 | 8,881 ms |
ジャッジサーバーID (参考情報) |
judge14 / judge13 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1 ms
4,376 KB |
testcase_01 | AC | 1 ms
4,376 KB |
testcase_02 | AC | 1 ms
4,376 KB |
testcase_03 | AC | 118 ms
26,224 KB |
testcase_04 | AC | 117 ms
26,784 KB |
testcase_05 | AC | 114 ms
25,260 KB |
testcase_06 | AC | 76 ms
18,708 KB |
testcase_07 | AC | 20 ms
7,612 KB |
testcase_08 | AC | 13 ms
5,976 KB |
testcase_09 | AC | 26 ms
8,964 KB |
testcase_10 | AC | 70 ms
17,852 KB |
testcase_11 | AC | 34 ms
10,736 KB |
testcase_12 | AC | 45 ms
12,360 KB |
testcase_13 | AC | 127 ms
27,380 KB |
testcase_14 | AC | 226 ms
27,280 KB |
testcase_15 | AC | 232 ms
27,400 KB |
testcase_16 | AC | 217 ms
27,360 KB |
testcase_17 | AC | 224 ms
27,504 KB |
testcase_18 | AC | 129 ms
20,696 KB |
testcase_19 | AC | 125 ms
22,236 KB |
testcase_20 | AC | 127 ms
20,452 KB |
testcase_21 | AC | 90 ms
16,772 KB |
testcase_22 | AC | 165 ms
27,244 KB |
testcase_23 | AC | 2 ms
4,380 KB |
testcase_24 | AC | 2 ms
4,380 KB |
testcase_25 | AC | 2 ms
4,376 KB |
testcase_26 | AC | 1 ms
4,376 KB |
testcase_27 | AC | 2 ms
4,380 KB |
testcase_28 | AC | 66 ms
20,904 KB |
testcase_29 | AC | 28 ms
8,540 KB |
testcase_30 | AC | 87 ms
28,128 KB |
testcase_31 | AC | 124 ms
28,136 KB |
testcase_32 | AC | 105 ms
28,052 KB |
testcase_33 | AC | 82 ms
27,348 KB |
testcase_34 | AC | 85 ms
27,220 KB |
testcase_35 | AC | 107 ms
27,276 KB |
testcase_36 | AC | 108 ms
27,304 KB |
testcase_37 | AC | 83 ms
27,332 KB |
ソースコード
#include <bits/stdc++.h> #pragma GCC optimize("Ofast") using namespace std; using std::cout; using std::cin; using std::endl; using ll=long long; using ld=long double; ll ILL=2167167167167167167; const int INF=2100000000; const int mod=998244353; #define rep(i,a,b) for (ll i=a;i<b;i++) #define all(p) p.begin(),p.end() template<class T> using _pq = priority_queue<T, vector<T>, greater<T>>; template<class T> ll LB(vector<T> &v,T a){return lower_bound(v.begin(),v.end(),a)-v.begin();} template<class T> ll UB(vector<T> &v,T a){return upper_bound(v.begin(),v.end(),a)-v.begin();} template<class T> bool chmin(T &a,const T &b){if(a>b){a=b;return 1;}else return 0;} template<class T> bool chmax(T &a,const T &b){if(a<b){a=b;return 1;}else return 0;} template<class T> void So(vector<T> &v) {sort(v.begin(),v.end());} template<class T> void Sore(vector<T> &v) {sort(v.begin(),v.end(),[](T x,T y){return x>y;});} void yneos(bool a){if(a) cout<<"Yes\n"; else cout<<"No\n";} template<class T> void vec_out(vector<T> &p){for(int i=0;i<(int)(p.size());i++){if(i) cout<<" ";cout<<p[i];}cout<<"\n";} template<class T> T min(vector<T> &a){assert(!a.empty());T ans=a[0];for(auto &x:a) chmin(ans,x);return ans;} template<class T> T max(vector<T> &a){assert(!a.empty());T ans=a[0];for(auto &x:a) chmax(ans,x);return ans;} template<class T> T sum(vector<T> &a){assert(!a.empty());T ans=a[0]-a[0];for(auto &x:a) ans+=x;return ans;} std::vector<std::vector<int>> tree_in(int N){ std::vector<std::vector<int>> G(N); for(int i=0;i<N-1;i++){ int a;int b; cin>>a>>b; a--,b--; G[a].push_back(b); G[b].push_back(a); } return G; } std::tuple<std::vector<int>,std::vector<int>,std::vector<int>> tree_order_pare_depth(std::vector<std::vector<int>> &G,int root){ int n=G.size(); std::vector<int> order={root},pare(n,-1),depth(n); pare[root]=-2; for(int i=0;i<n;i++){ int a=order[i]; for(auto x:G[a]){ if(pare[x]==-1){ pare[x]=a; depth[x]=depth[a]+1; order.push_back(x); } } } return {order,pare,depth}; } std::vector<int> tree_diameter_path(std::vector<std::vector<int>> &G){ int n=G.size(); auto r=(std::get<0>(tree_order_pare_depth(G,0))).at(n-1); std::vector<int> order,pare,depth,ans; tie(order,pare,depth)=tree_order_pare_depth(G,r); int ind=order[n-1]; while(ind!=-2){ ans.push_back(ind); ind=pare[ind]; } return ans; } void solve(); // oddloop int main() { ios::sync_with_stdio(false); cin.tie(nullptr); int t=1; //cin>>t; rep(i,0,t) solve(); } void solve(){ int N; cin>>N; auto G=tree_in(N); vector<int> order,pare,depth; tie(order,pare,depth)=tree_order_pare_depth(G,0); vector<vector<ll>> dp(N,vector<ll>(2)); for(int i=0;i<N;i++){ int a=order[N-i-1]; dp[a][0]=1; for(auto x:G[a]){ if(pare[a]==x) continue; ll tmp1=dp[a][0]*dp[x][0]; ll tmp2=dp[a][1]*dp[x][0]+dp[a][0]*dp[x][1]; dp[a][0]=tmp1%mod; dp[a][1]=tmp2%mod; } dp[a][1]=(dp[a][0]+dp[a][1])%mod; dp[a][0]=(dp[a][0]+dp[a][1])%mod; } cout<<(dp[0][1])<<"\n"; }