สมมติว่ามีอาร์เรย์ขนาด n ที่ถูกเริ่มต้นด้วย 0 และเรามีค่า k เราจะทำการอัปเดต k ครั้ง แต่ละการดำเนินการจะถูกแทนด้วย triplet: [startIndex, endIndex, inc] ซึ่งเพิ่มค่า inc ให้แต่ละสมาชิกของซับอาร์เรย์ A[startIndex ... endIndex] (รวม startIndex และ endIndex) เราต้องหาอาร์เรย์ที่ปรับเปลี่ยนแล้วหลังจากดำเนินการ k ครั้งทั้งหมด
เช่น ถ้าอินพุทเป็น length = 5, updates = [[1,3,2],[2,4,3],[0,2,-2]] ผลลัพธ์จะเป็น [-2,0,3,5,3]
เพื่อแก้ปัญหานี้ เราจะทำตามขั้นตอนดังนี้ −
กำหนดอาร์เรย์ ret ขนาด n
ลูป i := 0 ถึงขนาดของ a-1 ทำ −
l := a[i][0]
r := a[i][1] + 1
ret[l] := ret[l] + a[i][2]
ถ้า r < n แล้ว −
ret[r] := ret[r] - a[i][2]
ลูป i := 1 ถึง n-1 ทำ −
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, ]
