結果
問題 | No.2004 Incremental Coins |
ユーザー | woodywoody |
提出日時 | 2022-07-08 23:58:29 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 3,460 bytes |
コンパイル時間 | 4,206 ms |
コンパイル使用メモリ | 235,412 KB |
実行使用メモリ | 32,700 KB |
最終ジャッジ日時 | 2024-06-09 04:37:08 |
合計ジャッジ時間 | 12,622 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
5,248 KB |
testcase_01 | AC | 3 ms
5,376 KB |
testcase_02 | AC | 3 ms
5,376 KB |
testcase_03 | WA | - |
testcase_04 | WA | - |
testcase_05 | WA | - |
testcase_06 | WA | - |
testcase_07 | WA | - |
testcase_08 | RE | - |
testcase_09 | WA | - |
testcase_10 | RE | - |
testcase_11 | WA | - |
testcase_12 | WA | - |
testcase_13 | WA | - |
testcase_14 | WA | - |
testcase_15 | WA | - |
testcase_16 | WA | - |
testcase_17 | WA | - |
testcase_18 | WA | - |
testcase_19 | RE | - |
testcase_20 | WA | - |
testcase_21 | RE | - |
ソースコード
#include<bits/stdc++.h> #include<atcoder/all> #define rep(i,b) for(int i=0;i<b;i++) #define rrep(i,b) for(int i=b-1;i>=0;i--) #define rep1(i,b) for(int i=1;i<b;i++) #define repx(i,x,b) for(int i=x;i<b;i++) #define rrepx(i,x,b) for(int i=b-1;i>=x;i--) #define fore(i,a) for(auto i:a) #define fore1(i,a) for(auto &i:a) #define rng(x) (x).begin(), (x).end() #define rrng(x) (x).rbegin(), (x).rend() #define sz(x) ((int)(x).size()) #define pb push_back #define fi first #define se second using namespace std; using namespace atcoder; using ll = long long; template<typename T> using mpq = priority_queue<T, vector<T>, greater<T>>; template<typename T> bool chmax(T &a, const T &b) { if (a<b) { a=b; return 1; } return 0; } template<typename T> bool chmin(T &a, const T &b) { if (b<a) { a=b; return 1; } return 0; } template<typename T> ll sumv(const vector<T>&a){ll res(0);for(auto&&x:a)res+=x;return res;} bool yn(bool a) { if(a) {cout << "Yes" << endl; return 1;} else {cout << "No" << endl; return 0;}} #define dame { cout << "No" << endl; return 0;} #define dame1 { cout << -1 << endl; return 0;} #define test(x) cout << "test" << x << endl; #define deb(x,y) cout << x << " " << y << endl; #define deb3(x,y,z) cout << x << " " << y << " " << z << endl; #define deb4(x,y,z,x2) cout << x << " " << y << " " << z << " " << x2 << endl; #define out cout << ans << endl; #define outv fore(yans , ans) cout << yans << "\n"; #define show(x) cerr<<#x<<" = "<<x<<endl; #define showv(v) {fore(vy , v) {cout << vy << " ";} cout << endl;} #define showv2(v) fore(vy , v) cout << vy << "\n"; using pll = pair<ll,ll>; using pil = pair<int,ll>; using pli = pair<ll,int>; using pii = pair<int,int>; using tp = tuple<int ,int ,int>; using vi = vector<int>; using vl = vector<ll>; using vs = vector<string>; using vb = vector<bool>; using vpii = vector<pii>; using vpli = vector<pli>; using vpll = vector<pll>; using vpil = vector<pil>; using vvi = vector<vector<int>>; using vvl = vector<vector<ll>>; using vvs = vector<vector<string>>; using vvb = vector<vector<bool>>; using vvpii = vector<vector<pii>>; using vvpli = vector<vector<pli>>; using vvpll = vector<vpll>; using vvpil = vector<vpil>; using mint = modint998244353; using vm = vector<mint>; using vvm = vector<vector<mint>>; vector<int> dx={1,0,-1,0,1,1,-1,-1},dy={0,1,0,-1,1,-1,1,-1}; ll gcd(ll a, ll b) { return a?gcd(b%a,a):b;} ll lcm(ll a, ll b) { return a/gcd(a,b)*b;} const double eps = 1e-10; const ll LINF = 1001002003004005006ll; const int INF = 1001001001; vvi e; int n; ll k; vm sum; vm ans; vl a; vm d; mint dfs(int x,int pre,int dep){ mint tmp = 0; mint tmp_sum = 0; mint tmp_d = 0; int nd = dep+k+1; if (nd < n) tmp_d -= sum[nd]; fore(y , e[x]){ if(y == pre) continue; tmp += dfs(y , x , dep+1); tmp_sum += ans[y] + mint(a[y]) * mint(k); } if (nd < n) tmp_d += sum[nd]; ans[x] = tmp_sum - tmp + a[x]; sum[dep] += ans[x]; tmp -= tmp_d; tmp += a[x]; return tmp; } int main(){ cin>>n>>k; n++; e.resize(n); sum.resize(n); ans.resize(n); a.resize(n); d.resize(n); vl p(n); rep(i,n) cin>>a[i]; rep1(i,n) cin>>p[i]; rep1(i,n){ e[i].pb(p[i]); e[p[i]].pb(i); } dfs(0 , -1 , 0); // mint all = 0; // rep(i,n) all += a[i]; // rep1(i,n) all -= ans[i]; // ans[0] = all; rep(i,n) cout << ans[i].val() << endl; return 0; }