#include <bits/stdc++.h>
using namespace std;
int main(){
ios_base::sync_with_stdio(false);
cin.tie(0);
int k, n0;
cin >> k >> n0;
if(k == 1){
cout << n0;
return 0;
}
vector<vector<pair<int, long long int>>> tab(k);
// a, ile musi przyjsc (jesli 0 -> 1)
tab[0] = vector<pair<int, long long int>>(n0, {0, 0});
for(int i=1; i<k; i++){
int n;
cin >> n;
for(int j=0; j<n; j++){
int a;
cin >> a;
tab[i].push_back({a, 0});
}
}
long long int ans = 0;
for(int i=k-1; i>=0; i--){
for(int j=0; j<int(tab[i].size()); j++){
if(tab[i][j].first != 0){
tab[i-1][tab[i][j].first-1].second += max(1ll, tab[i][j].second);
}
if(i == 0) ans += max(1ll, tab[i][j].second);
// cout << i << " " << j << " " << tab[i][j].second << "\n";
}
}
cout << ans;
}
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 | #include <bits/stdc++.h> using namespace std; int main(){ ios_base::sync_with_stdio(false); cin.tie(0); int k, n0; cin >> k >> n0; if(k == 1){ cout << n0; return 0; } vector<vector<pair<int, long long int>>> tab(k); // a, ile musi przyjsc (jesli 0 -> 1) tab[0] = vector<pair<int, long long int>>(n0, {0, 0}); for(int i=1; i<k; i++){ int n; cin >> n; for(int j=0; j<n; j++){ int a; cin >> a; tab[i].push_back({a, 0}); } } long long int ans = 0; for(int i=k-1; i>=0; i--){ for(int j=0; j<int(tab[i].size()); j++){ if(tab[i][j].first != 0){ tab[i-1][tab[i][j].first-1].second += max(1ll, tab[i][j].second); } if(i == 0) ans += max(1ll, tab[i][j].second); // cout << i << " " << j << " " << tab[i][j].second << "\n"; } } cout << ans; } |
English