結果

問題 No.2214 Products on Tree
ユーザー Taiki0715Taiki0715
提出日時 2024-02-01 14:46:28
言語 C++17(clang)
(17.0.6 + boost 1.83.0)
結果
AC  
実行時間 218 ms / 3,000 ms
コード長 3,047 bytes
コンパイル時間 7,800 ms
コンパイル使用メモリ 178,376 KB
実行使用メモリ 65,768 KB
最終ジャッジ日時 2024-02-01 14:46:45
合計ジャッジ時間 15,468 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge14
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,676 KB
testcase_01 AC 2 ms
6,676 KB
testcase_02 AC 1 ms
6,676 KB
testcase_03 AC 178 ms
24,220 KB
testcase_04 AC 175 ms
24,612 KB
testcase_05 AC 158 ms
23,512 KB
testcase_06 AC 126 ms
17,428 KB
testcase_07 AC 23 ms
7,424 KB
testcase_08 AC 15 ms
6,676 KB
testcase_09 AC 33 ms
8,704 KB
testcase_10 AC 127 ms
16,640 KB
testcase_11 AC 47 ms
10,368 KB
testcase_12 AC 62 ms
11,776 KB
testcase_13 AC 191 ms
25,284 KB
testcase_14 AC 177 ms
25,288 KB
testcase_15 AC 218 ms
25,288 KB
testcase_16 AC 185 ms
25,284 KB
testcase_17 AC 186 ms
25,288 KB
testcase_18 AC 105 ms
20,112 KB
testcase_19 AC 109 ms
21,312 KB
testcase_20 AC 96 ms
19,804 KB
testcase_21 AC 72 ms
16,128 KB
testcase_22 AC 141 ms
26,044 KB
testcase_23 AC 2 ms
6,676 KB
testcase_24 AC 2 ms
6,676 KB
testcase_25 AC 1 ms
6,676 KB
testcase_26 AC 2 ms
6,676 KB
testcase_27 AC 2 ms
6,676 KB
testcase_28 AC 92 ms
48,952 KB
testcase_29 AC 36 ms
8,704 KB
testcase_30 AC 80 ms
27,380 KB
testcase_31 AC 157 ms
27,380 KB
testcase_32 AC 151 ms
27,308 KB
testcase_33 AC 139 ms
65,768 KB
testcase_34 AC 123 ms
65,768 KB
testcase_35 AC 163 ms
45,444 KB
testcase_36 AC 159 ms
45,444 KB
testcase_37 AC 97 ms
25,064 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#if __has_include(<atcoder/all>)
#include <atcoder/all>
using namespace atcoder;
template<int mod>istream &operator>>(istream &is,static_modint<mod> &a){long long b;is>>b;a=b;return is;}
istream &operator>>(istream &is,modint &a){long long b;cin>>b;a=b;return is;}
#endif
#ifdef LOCAL
#include "debug.h"
#else
#define debug(...) static_cast<void>(0)
#define debugg(...) static_cast<void>(0)
template<typename T1,typename T2>ostream &operator<<(ostream &os,const pair<T1,T2>&p){os<<p.first<<' '<<p.second;return os;}
#endif
using ll=long long;
using ull=unsigned long long;
using P=pair<ll,ll>;
template<typename T>using minque=priority_queue<T,vector<T>,greater<T>>;
template<typename T>bool chmax(T &a,const T &b){return (a<b?(a=b,true):false);}
template<typename T>bool chmin(T &a,const T &b){return (a>b?(a=b,true):false);}
template<typename T1,typename T2>istream &operator>>(istream &is,pair<T1,T2>&p){is>>p.first>>p.second;return is;}
template<typename T>istream &operator>>(istream &is,vector<T> &a){for(auto &i:a)is>>i;return is;}
template<typename T1,typename T2>void operator++(pair<T1,T2>&a,int n){a.first++,a.second++;}
template<typename T1,typename T2>void operator--(pair<T1,T2>&a,int n){a.first--,a.second--;}
template<typename T>void operator++(vector<T>&a,int n){for(auto &i:a)i++;}
template<typename T>void operator--(vector<T>&a,int n){for(auto &i:a)i--;}
#define reps(i,a,n) for(int i=(a);i<(n);i++)
#define rep(i,n) reps(i,0,n)
#define all(x) x.begin(),x.end()
#define pcnt(x) __builtin_popcountll(x)
ll myceil(ll a,ll b){return (a+b-1)/b;}
template<typename T,size_t n,size_t id=0>
auto vec(const int (&d)[n],const T &init=T()){
  if constexpr (id<n)return vector(d[id],vec<T,n,id+1>(d,init));
  else return init;
}
void SOLVE();
int main(){
  ios::sync_with_stdio(false);
  cin.tie(nullptr);
  #ifdef LOCAL
  clock_t start=clock();
  #endif
  int testcase=1;
  //cin>>testcase;
  for(int i=0;i<testcase;i++){
    SOLVE();
  }
  #ifdef LOCAL
  cerr<<"time:";
  cerr<<(clock()-start)/1000;
  cerr<<"ms\n";
  #endif
}
void SOLVE(){
  int n;
  cin>>n;
  vector<vector<int>>to(n);
  rep(i,n-1){
    int u,v;
    cin>>u>>v;
    u--,v--;
    to[u].push_back(v);
    to[v].push_back(u);
  }
  using mint=modint998244353;
  vector<vector<mint>>dp(n,vector<mint>(2,0));
  auto dfs=[&](auto self,int x,int p)->void {
    dp[x][0]=dp[x][1]=1;
    vector<mint>left(to[x].size()),right(to[x].size());
    for(auto i:to[x])if(i!=p){
      self(self,i,x);
      dp[x][0]*=dp[i][0]+dp[i][1];
      dp[x][1]*=dp[i][0]+dp[i][1];
    }
    rep(i,to[x].size()){
      if(to[x][i]!=p)left[i]=right[i]=dp[to[x][i]][0]+dp[to[x][i]][1];
      else left[i]=right[i]=1;
    }
    reps(i,1,left.size())left[i]*=left[i-1];
    for(int i=left.size()-1;i>=1;i--)right[i-1]*=right[i];
    rep(i,to[x].size())if(to[x][i]!=p){
      mint now=1;
      if(i)now*=left[i-1];
      if(i+1!=to[x].size())now*=right[i+1];
      dp[x][1]+=dp[to[x][i]][1]*now;
    }
  };
  dfs(dfs,0,-1);
  cout<<dp[0][1].val()<<endl;
}
0