#include <bits/stdc++.h>
using namespace std;
#define pb push_back
#define tak cout<<"TAK\n"
#define yes cout<<"YES\n"
#define nie cout<<"NIE\n"
#define no cout<<"NO\n"
using ll = long long;
using ld = long double;
using pll = pair<ll,ll>;
using pdd = pair<ld,ld>;
#define vec vector
#define all(x) (x).begin(), (x).end()
#define sz(x) (x).size()
#define linijki ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
#define mapa map<ll,ll>
#define umapa unordered_map<ll,ll>
#define se set<ll>
#define ff first
#define ss second
#define str string
const ll alph = 26;
const ll BIN6 = (1LL << 20);
const ll BIN9 = (1LL << 30);
const ll BIN18 = (1LL << 60);
const ll INF5 = 100000;
const ll INF6 = 1000000;
const ll INF9 = 1000000000;
const ll INF18 = 1000000000000000000;
const ld DINF5 = 1e5L;
const ld DINF6 = 1e6L;
const ld DINF9 = 1e9L;
const ll MOD = 1000000007;
int main() {
linijki;
ll k,n;
cin>>k>>n;
vec<ll> roz(k);
roz[0]=n;
vec<vec<ll>> spo(k);
spo[0].resize(n + 1, 0);
for(ll i = 1; i < k; i++){
cin>>roz[i];
spo[i].resize(roz[i] + 1);
for(ll j=1; j<=roz[i]; j++){
cin>>spo[i][j];
}
}
vec<vec<ll>> minim(k);
for(ll i = 0; i < k; i++){
minim[i].resize(roz[i] + 1, 0);
}
/////////////////////////////////////////////////////////////////////////
for(ll i=k-1; i>= 1;i--){
for(ll j = 1; j<=roz[i]; j++){
minim[i][j]=max(minim[i][j], 1LL);
if(spo[i][j]>0){
minim[i-1][spo[i][j]]+=minim[i][j];
}
}
}
/*
for(ll i=1; i<k;i++){
for(ll j = 1; j<=roz[i]; j++){
cout<<minim[i][j];
}
}
*/
////////////////////////////////////////////////////////////////////////////////////////
ll jeden=1,wyn=0, aktywni=0;
for(ll i=1;i<=n; i++){
minim[0][i]=max(minim[0][i],jeden);
}
for(ll i=0;i< k; i++){
ll zero=0;
vec<ll> sum(roz[i]+1, zero);
if((i+1)<k){
for(ll j=1; j<=roz[i+1];j++){
if(spo[i+1][j]>(0)){
sum[spo[i+1][j]]+=minim[i+1][j];
}
}
}
ll potrzebni_na_start=0;
for(ll j =1; j<= roz[i];j++){
if(i == 0 or spo[i][j] == 0){
potrzebni_na_start+=minim[i][j];
}
}
ll uzyjemy = min(potrzebni_na_start, aktywni);
wyn +=(potrzebni_na_start-uzyjemy);
aktywni-=uzyjemy;
ll przechodzi_dalej=0;
for(ll j = 1; j <= roz[i]; j++){
przechodzi_dalej+=(minim[i][j]-sum[j]);
}
aktywni+=przechodzi_dalej;
}
cout<<wyn;
return 0;
}
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 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 | #include <bits/stdc++.h> using namespace std; #define pb push_back #define tak cout<<"TAK\n" #define yes cout<<"YES\n" #define nie cout<<"NIE\n" #define no cout<<"NO\n" using ll = long long; using ld = long double; using pll = pair<ll,ll>; using pdd = pair<ld,ld>; #define vec vector #define all(x) (x).begin(), (x).end() #define sz(x) (x).size() #define linijki ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); #define mapa map<ll,ll> #define umapa unordered_map<ll,ll> #define se set<ll> #define ff first #define ss second #define str string const ll alph = 26; const ll BIN6 = (1LL << 20); const ll BIN9 = (1LL << 30); const ll BIN18 = (1LL << 60); const ll INF5 = 100000; const ll INF6 = 1000000; const ll INF9 = 1000000000; const ll INF18 = 1000000000000000000; const ld DINF5 = 1e5L; const ld DINF6 = 1e6L; const ld DINF9 = 1e9L; const ll MOD = 1000000007; int main() { linijki; ll k,n; cin>>k>>n; vec<ll> roz(k); roz[0]=n; vec<vec<ll>> spo(k); spo[0].resize(n + 1, 0); for(ll i = 1; i < k; i++){ cin>>roz[i]; spo[i].resize(roz[i] + 1); for(ll j=1; j<=roz[i]; j++){ cin>>spo[i][j]; } } vec<vec<ll>> minim(k); for(ll i = 0; i < k; i++){ minim[i].resize(roz[i] + 1, 0); } ///////////////////////////////////////////////////////////////////////// for(ll i=k-1; i>= 1;i--){ for(ll j = 1; j<=roz[i]; j++){ minim[i][j]=max(minim[i][j], 1LL); if(spo[i][j]>0){ minim[i-1][spo[i][j]]+=minim[i][j]; } } } /* for(ll i=1; i<k;i++){ for(ll j = 1; j<=roz[i]; j++){ cout<<minim[i][j]; } } */ //////////////////////////////////////////////////////////////////////////////////////// ll jeden=1,wyn=0, aktywni=0; for(ll i=1;i<=n; i++){ minim[0][i]=max(minim[0][i],jeden); } for(ll i=0;i< k; i++){ ll zero=0; vec<ll> sum(roz[i]+1, zero); if((i+1)<k){ for(ll j=1; j<=roz[i+1];j++){ if(spo[i+1][j]>(0)){ sum[spo[i+1][j]]+=minim[i+1][j]; } } } ll potrzebni_na_start=0; for(ll j =1; j<= roz[i];j++){ if(i == 0 or spo[i][j] == 0){ potrzebni_na_start+=minim[i][j]; } } ll uzyjemy = min(potrzebni_na_start, aktywni); wyn +=(potrzebni_na_start-uzyjemy); aktywni-=uzyjemy; ll przechodzi_dalej=0; for(ll j = 1; j <= roz[i]; j++){ przechodzi_dalej+=(minim[i][j]-sum[j]); } aktywni+=przechodzi_dalej; } cout<<wyn; return 0; } |
English