C++での範囲の追加

サイズnの配列があり、それが0で初期化され、値kもあるとすると、k個の更新操作を実行します。各操作はトリプレットとして表されます：[startIndex、endIndex、inc]これはサブ配列A [startIndex ... endIndex]（startIndexおよびendIndexを含む）の各要素をincでインクリメントします。 k個の操作がすべて実行された後、変更された配列を見つける必要があります。

したがって、入力が長さ=5、更新=[[1,3,2]、[2,4,3]、[0,2、-2]]のような場合、出力は[-2,0]になります。 、3,5,3]

これを解決するには、次の手順に従います-

• サイズnの配列retを定義します

• 初期化i：=0の場合、i

  • l：=a [i、0]

  • r：=a [i、1] + 1

  • ret [l]：=ret [l] + a [i、2]

  • r

    • ret [r]：=ret [r]-a [i、2]

• 初期化i：=1の場合、i

  • ret [i]：=ret [i] + ret [i-1]

• retを返す

例

理解を深めるために、次の実装を見てみましょう-

#include <bits/stdc++.h>
using namespace std;
void print_vector(vector<auto< v){
   cout << "[";
   for(int i = 0; i<v.size(); i++){
      cout << v[i] << ", ";
   }
   cout << "]"<<endl;
}
class Solution {
public:
   vector<int< getModifiedArray(int n, vector& a) {
      vector<int< ret(n);
      for (int i = 0; i < a.size(); i++) {
         int l = a[i][0];
         int r = a[i][1] + 1;
         ret[l] += a[i][2];
         if (r < n) {
            ret[r] -= a[i][2];
         }
      }
      for (int i = 1; i < n; i++) {
         ret[i] += ret[i - 1];
      }
      return ret;
   }
};
main(){
   Solution ob;
   vector<vector<int<> v = {{1,3,2},{2,4,3},{0,2,-2}};
   print_vector(ob.getModifiedArray(5,v));
}

入力

5, {{1,3,2},{2,4,3},{0,2,-2}}

出力

[-2, 0, 3, 5, 3, ]