Phép cộng phạm vi trong C ++

Giả sử chúng ta có một mảng kích thước n và được khởi tạo bằng các số 0 và chúng ta cũng có giá trị k, chúng ta sẽ thực hiện k thao tác cập nhật. Mỗi hoạt động sẽ được biểu diễn dưới dạng bộ ba:[startIndex, endIndex, inc] làm tăng từng phần tử của mảng con A [startIndex ... endIndex] (startIndex và endIndex inclusive) với inc. Chúng ta phải tìm mảng đã sửa đổi sau khi tất cả k hoạt động được thực thi.

Vì vậy, nếu đầu vào là length =5, Updates =[[1,3,2], [2,4,3], [0,2, -2]], thì đầu ra sẽ là [- 2,0 , 3,5,3]

Để giải quyết vấn đề này, chúng tôi sẽ làm theo các bước sau -

• Xác định ret mảng có kích thước n

• để khởi tạo i:=0, khi i

  • l:=a [i, 0]

  • r:=a [i, 1] + 1

  • ret [l]:=ret [l] + a [i, 2]

  • nếu r

    • ret [r]:=ret [r] - a [i, 2]

• để khởi tạo i:=1, khi i

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

• trả lại ret

Ví dụ

Hãy cùng chúng tôi xem cách triển khai sau để hiểu rõ hơn -

#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));
}

Đầu vào

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

Đầu ra

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