_takumi

First

Dec 21st, 2019
385
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 0.93 KB | None | 0 0
  1. #include <vector>
  2. #include <iostream>
  3. #include <algorithm>
  4. using namespace std;
  5.  
  6. int main() {
  7.     int m, n, k;
  8.     cin >> m >> n >> k;
  9.     vector<int> a;
  10.     for (int i = 0; i < m; ++i) {
  11.         cin >> a[i];
  12.     }
  13.     vector<vector<int>> v;
  14.     for (int i = 0; i < n; ++i) {
  15.         for (int j = 0; j < k; ++j) {
  16.             cin >> v[i][j];
  17.         }
  18.     }
  19.     reverse(begin(v), end(v));
  20.     int curr_flat = 0;
  21.     int c = 0;
  22.     int lit = 0;
  23.     vector<int> flat_indices;
  24.     for (int i = 0; i < n; ++i) {
  25.         for (size_t j = 0; j < a.size(); ++j) {
  26.             for (size_t b = 0; b < a[j]; ++b, lit += v[i][b]);
  27.         }
  28.         if (lit == 0)
  29.             flat_indices.push_back(m * i + curr_flat + 1);
  30.     }
  31.     cout << flat_indices.size() << "\n";
  32.     for (int i = 0; i < flat_indices.size(); ++i) {
  33.         cout << flat_indices[i] << " ";
  34.     }
  35.     return 0;
  36.     /*for (int j = 0; j < k; ++j, ++c, lit += v[i][j]) {
  37.         if (k == a[curr_flat]) {
  38.             curr_flat++;
  39.             c = 0;
  40.             if (lit == 0)
  41.                 flat_indices.push_back();
  42.         }
  43.     }*/
  44. }
Advertisement
Add Comment
Please, Sign In to add comment