การเพิ่มช่วงใน C++

สมมติว่าเรามีอาร์เรย์ขนาด n และที่เริ่มต้นด้วย 0 และเรามีค่า k ด้วย เราจะดำเนินการอัปเดต k การดำเนินการแต่ละรายการจะแสดงเป็น triplet:[startIndex, endIndex, inc] ซึ่งเพิ่มแต่ละองค์ประกอบของ subarray A[startIndex ... endIndex] (รวม startIndex และ endIndex) ด้วย inc เราต้องหาอาร์เรย์ที่แก้ไขหลังจากดำเนินการ k ทั้งหมดแล้ว

ดังนั้น หากอินพุตเท่ากับ length =5, updates =[[1,3,2],[2,4,3],[0,2,-2]] ผลลัพธ์จะเป็น [- 2,0 ,3,5,3]

เพื่อแก้ปัญหานี้ เราจะทำตามขั้นตอนเหล่านี้ -

• กำหนดขนาดของอาร์เรย์ n

• สำหรับการเริ่มต้น 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 เมื่อฉัน

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

• รีเทิร์น

ตัวอย่าง

ให้เราดูการใช้งานต่อไปนี้เพื่อความเข้าใจที่ดีขึ้น -

#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, ]