#include <iostream>
#include <vector>
#include <numeric>
typedef struct conf {
uint32_t rows;
std::vector<std::vector<uint32_t>> input;
std::vector<std::vector<uint32_t>> lectureCast;
uint32_t minCast;
};
void printCast(conf &c) {
std::cerr << "lectCast: \n";
for (uint32_t i = 0; i < c.lectureCast.size(); ++i) {
for (uint32_t j = 0; j< c.lectureCast[i].size(); ++j) {
std::cerr << c.lectureCast[i][j] << " ";
}
std::cerr << "\n";
}
}
uint32_t sumRow(std::vector<uint32_t> &v) {
uint32_t result = 0;
for(int32_t i =0; i< v.size(); ++i) {
if (v[i] ==0){
result +=1;
}else {
result += v[i];
}
}
return result;
}
void prog_main(std::istream& in, std::ostream& out)
{
uint32_t rows;
uint32_t rowSize;
in >> rows;
in >> rowSize;
conf c{
rows,
std::vector<std::vector<uint32_t>>{rows, std::vector<uint32_t>(rowSize, 0)},
std::vector<std::vector<uint32_t>>{rows, std::vector<uint32_t>(rowSize, 0)},
0
};
for (uint32_t i = 1; i < rows; ++i) {
in >> rowSize;
c.input[i] = std::vector<uint32_t>(rowSize, 0);
c.lectureCast[i] = std::vector<uint32_t>(rowSize, 0);
for (uint32_t j = 0; j < rowSize; ++j) {
in >> c.input[i][j];
}
}
/* std::cerr << "input:" << rows << "\n";
for (uint32_t i = 0; i < rows; ++i) {
for (uint32_t j = 0; j < c.input[i].size(); ++j) {
std::cerr << c.input[i][j] << " ";
}
std::cerr << "\n";
}
*/
//printCast(c);
for (int32_t i = rows - 1; i >= 0; i--) {
// std::cerr << "i:" << i << "\n";
for (uint32_t j = 0; j < c.input[i].size(); ++j) {
if (c.lectureCast[i][j] == 0 ) {
c.lectureCast[i][j] = 1;
}
uint32_t contOf = c.input[i][j];
if (contOf != 0) {
// std::cerr << "[" << i << "]" << "[" << contOf-1 << "]\n";
c.lectureCast[i-1][contOf-1] += c.lectureCast[i][j];
}
}
uint32_t castForRow = sumRow(c.lectureCast[i]);
if (castForRow > c.minCast) {
c.minCast = castForRow;
}
}
//printCast(c);
out << c.minCast;
}
#ifndef TEST
int main(int argc, char* argv[])
{
prog_main(std::cin, std::cout);
return 0;
}
#endif
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 | #include <iostream> #include <vector> #include <numeric> typedef struct conf { uint32_t rows; std::vector<std::vector<uint32_t>> input; std::vector<std::vector<uint32_t>> lectureCast; uint32_t minCast; }; void printCast(conf &c) { std::cerr << "lectCast: \n"; for (uint32_t i = 0; i < c.lectureCast.size(); ++i) { for (uint32_t j = 0; j< c.lectureCast[i].size(); ++j) { std::cerr << c.lectureCast[i][j] << " "; } std::cerr << "\n"; } } uint32_t sumRow(std::vector<uint32_t> &v) { uint32_t result = 0; for(int32_t i =0; i< v.size(); ++i) { if (v[i] ==0){ result +=1; }else { result += v[i]; } } return result; } void prog_main(std::istream& in, std::ostream& out) { uint32_t rows; uint32_t rowSize; in >> rows; in >> rowSize; conf c{ rows, std::vector<std::vector<uint32_t>>{rows, std::vector<uint32_t>(rowSize, 0)}, std::vector<std::vector<uint32_t>>{rows, std::vector<uint32_t>(rowSize, 0)}, 0 }; for (uint32_t i = 1; i < rows; ++i) { in >> rowSize; c.input[i] = std::vector<uint32_t>(rowSize, 0); c.lectureCast[i] = std::vector<uint32_t>(rowSize, 0); for (uint32_t j = 0; j < rowSize; ++j) { in >> c.input[i][j]; } } /* std::cerr << "input:" << rows << "\n"; for (uint32_t i = 0; i < rows; ++i) { for (uint32_t j = 0; j < c.input[i].size(); ++j) { std::cerr << c.input[i][j] << " "; } std::cerr << "\n"; } */ //printCast(c); for (int32_t i = rows - 1; i >= 0; i--) { // std::cerr << "i:" << i << "\n"; for (uint32_t j = 0; j < c.input[i].size(); ++j) { if (c.lectureCast[i][j] == 0 ) { c.lectureCast[i][j] = 1; } uint32_t contOf = c.input[i][j]; if (contOf != 0) { // std::cerr << "[" << i << "]" << "[" << contOf-1 << "]\n"; c.lectureCast[i-1][contOf-1] += c.lectureCast[i][j]; } } uint32_t castForRow = sumRow(c.lectureCast[i]); if (castForRow > c.minCast) { c.minCast = castForRow; } } //printCast(c); out << c.minCast; } #ifndef TEST int main(int argc, char* argv[]) { prog_main(std::cin, std::cout); return 0; } #endif |
English