SRM 576, Division 2 Level1

TheExperimentDiv2.cpp

#include <vector>
using namespace std;

class TheExperimentDiv2 {

  public: vector<int> determineHumidity(vector<int> intensity, int L, vector<int> leftEnd) {

    int N = (int)intensity.size();
    int M = (int)leftEnd.size();

    bool drop_flag[N];
    vector<int>absorbed;

    int i, j, k;

    for(i=0; i<N; i++) {
      drop_flag[i] = false;
    }

    for(i=0; i<M; i++) {
      int temp = 0;

      for(j=0; j<L; j++) {
        k = leftEnd[i]+j;

        if( drop_flag[k] == 0 ) {
          temp += intensity[k];
          drop_flag[k] = true;
        }
      }
      absorbed.push_back(temp);
    }

    return absorbed;
  }

};