C++ में रेंज एडिशन - अंतर सरणी तकनीक

समस्या विवरण

मान लीजिए हमारे पास आकार n की एक सरणी है जो 0 से इनिशियलाइज़ है और हमारे पास एक मान k है, हम k अपडेट ऑपरेशन करेंगे। प्रत्येक ऑपरेशन को ट्रिपलेट के रूप में दर्शाया जाएगा: [startIndex, endIndex, inc] जो सबएरे A[startIndex ... endIndex] (startIndex और endIndex समावेशी) के प्रत्येक तत्व को inc से बढ़ाता है। हमें सभी k ऑपरेशन निष्पादित होने के बाद संशोधित सरणी ढूंढनी है।

उदाहरण के लिए, यदि इनपुट length = 5, updates = [[1,3,2],[2,4,3],[0,2,-2]] है, तो आउटपुट [-2,0,3,5,3] होगा।

समाधान दृष्टिकोण

अंतर सरणी (Difference Array) तकनीक का उपयोग करके O(n + k) समय में:

1. आकार n की एक सरणी ret परिभाषित करें
2. प्रत्येक अपडेट के लिए:
   • 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]
3. प्रत्येक i के लिए 1 से n-1 तक: ret[i] := ret[i] + ret[i-1]
4. ret लौटाएं

C++ कार्यान्वयन

#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<vector<int>>& 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; 
    } 
}; 

int main(){ 
    Solution ob; 
    vector<vector<int>> v = {{1,3,2},{2,4,3},{0,2,-2}}; 
    print_vector(ob.getModifiedArray(5,v)); 
    return 0;
}

इनपुट

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

आउटपुट

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