#include <iostream>
#include <vector>
using namespace std;
int main() {
ios::sync_with_stdio(0); cin.tie(0);
int k, n1;
if (!(cin >> k >> n1)) return 0;
// dp[i][j] to liczba pracowników potrzebna na j-tym spotkaniu i-tego dnia
vector<vector<long long>> dp(k + 1);
vector<vector<int>> rodzic(k + 1);
dp[1].assign(n1 + 1, 0);
for (int i = 2; i <= k; ++i) {
int ni; cin >> ni;
dp[i].assign(ni + 1, 0);
rodzic[i].assign(ni + 1, 0);
for (int j = 1; j <= ni; ++j) cin >> rodzic[i][j];
}
long long wynik = 0;
// Lecimy od tyłu (od ostatniego dnia do pierwszego)
for (int i = k; i >= 1; --i) {
for (int j = 1; j < (int)dp[i].size(); ++j) {
// Jeśli nikt tu nie przyszedł z dnia i+1, to spotkanie potrzebuje min. 1 osoby
if (dp[i][j] == 0) dp[i][j] = 1;
if (i > 1 && rodzic[i][j] > 0) {
// To spotkanie jest kontynuacją, więc "ciągnie" ludzi z dnia wcześniej
dp[i - 1][rodzic[i][j]] += dp[i][j];
} else {
// To spotkanie NIE jest kontynuacją, więc musimy tu przysłać nowych ludzi
wynik += dp[i][j];
}
}
}
cout << wynik << endl;
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 | #include <iostream> #include <vector> using namespace std; int main() { ios::sync_with_stdio(0); cin.tie(0); int k, n1; if (!(cin >> k >> n1)) return 0; // dp[i][j] to liczba pracowników potrzebna na j-tym spotkaniu i-tego dnia vector<vector<long long>> dp(k + 1); vector<vector<int>> rodzic(k + 1); dp[1].assign(n1 + 1, 0); for (int i = 2; i <= k; ++i) { int ni; cin >> ni; dp[i].assign(ni + 1, 0); rodzic[i].assign(ni + 1, 0); for (int j = 1; j <= ni; ++j) cin >> rodzic[i][j]; } long long wynik = 0; // Lecimy od tyłu (od ostatniego dnia do pierwszego) for (int i = k; i >= 1; --i) { for (int j = 1; j < (int)dp[i].size(); ++j) { // Jeśli nikt tu nie przyszedł z dnia i+1, to spotkanie potrzebuje min. 1 osoby if (dp[i][j] == 0) dp[i][j] = 1; if (i > 1 && rodzic[i][j] > 0) { // To spotkanie jest kontynuacją, więc "ciągnie" ludzi z dnia wcześniej dp[i - 1][rodzic[i][j]] += dp[i][j]; } else { // To spotkanie NIE jest kontynuacją, więc musimy tu przysłać nowych ludzi wynik += dp[i][j]; } } } cout << wynik << endl; return 0; } |
English