#include <iostream>
#include <vector>
using namespace std;
int main()
{
int k;
cin >> k;
int* days = new int[k];
vector<vector<int>> conf(k);
vector<vector<int>> pers(k);
int n;
cin >> n;
days[0] = n;
conf[0].resize(n);
pers[0].resize(n);
for (int i = 0; i < n; i++) {
conf[0][i] = 0;
}
for (int i = 1; i < k; i++) {
cin >> n;
days[i] = n;
conf[i].resize(n);
pers[i].resize(n);
for (int j = 0; j < n; j++) {
cin >> conf[i][j];
}
}
/*
for (int i = 0; i < k; i++) {
cout << days[i] << " ";
for (int j = 0; j < days[i]; j++) {
cout << conf[i][j] << " ";
}
cout << endl;
}
cout << "=====" << endl;
*/
for (int j = 0; j < days[k - 1]; j++) {
pers[k - 1][j] = 0;
}
int sum = 0;
int sumMax = 0;
for (int i = k - 1; i > 0; i--) {
sum = 0;
for (int j = 0; j < days[i]; j++) {
if (pers[i][j] == 0)
pers[i][j] = 1;
sum += pers[i][j];
}
if (sum > sumMax) sumMax = sum;
for (int j = 0; j < days[i - 1]; j++) {
pers[i - 1][j] = 0;
}
for (int j = 0; j < days[i]; j++) {
if (conf[i][j] > 0)
pers[i - 1][conf[i][j] - 1] += pers[i][j];
}
}
sum = 0;
for (int j = 0; j < days[0]; j++) {
if (pers[0][j] == 0)
pers[0][j] = 1;
sum += pers[0][j];
}
if (sum > sumMax) sumMax = sum;
/*
for (int i = 0; i < k; i++) {
cout << days[i] << " ";
for (int j = 0; j < days[i]; j++) {
cout << pers[i][j] << " ";
}
cout << endl;
}
cout << "=====" << endl;
*/
cout << sumMax << endl;
}
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 | #include <iostream> #include <vector> using namespace std; int main() { int k; cin >> k; int* days = new int[k]; vector<vector<int>> conf(k); vector<vector<int>> pers(k); int n; cin >> n; days[0] = n; conf[0].resize(n); pers[0].resize(n); for (int i = 0; i < n; i++) { conf[0][i] = 0; } for (int i = 1; i < k; i++) { cin >> n; days[i] = n; conf[i].resize(n); pers[i].resize(n); for (int j = 0; j < n; j++) { cin >> conf[i][j]; } } /* for (int i = 0; i < k; i++) { cout << days[i] << " "; for (int j = 0; j < days[i]; j++) { cout << conf[i][j] << " "; } cout << endl; } cout << "=====" << endl; */ for (int j = 0; j < days[k - 1]; j++) { pers[k - 1][j] = 0; } int sum = 0; int sumMax = 0; for (int i = k - 1; i > 0; i--) { sum = 0; for (int j = 0; j < days[i]; j++) { if (pers[i][j] == 0) pers[i][j] = 1; sum += pers[i][j]; } if (sum > sumMax) sumMax = sum; for (int j = 0; j < days[i - 1]; j++) { pers[i - 1][j] = 0; } for (int j = 0; j < days[i]; j++) { if (conf[i][j] > 0) pers[i - 1][conf[i][j] - 1] += pers[i][j]; } } sum = 0; for (int j = 0; j < days[0]; j++) { if (pers[0][j] == 0) pers[0][j] = 1; sum += pers[0][j]; } if (sum > sumMax) sumMax = sum; /* for (int i = 0; i < k; i++) { cout << days[i] << " "; for (int j = 0; j < days[i]; j++) { cout << pers[i][j] << " "; } cout << endl; } cout << "=====" << endl; */ cout << sumMax << endl; } |
English