मान लीजिए हमारे पास एक m * n मैट्रिक्स है जिसे mat कहते हैं और एक इंटीजर K है, हमें एक और मैट्रिक्स answer ढूंढना है जहां प्रत्येक answer[i][j] उन सभी एलिमेंट्स mat[r][c] का योग है जिनके लिए i - K <= r <= i + K, j - K <= c <= j + K, और (r, c) मैट्रिक्स में एक वैध पोजीशन है। यदि इनपुट इस प्रकार है -
| 1 | 2 | 3 |
| 4 | 5 | 6 |
| 7 | 8 | 9 |
और k = 1 है, तो आउटपुट होगा -
| 12 | 21 | 16 |
| 27 | 45 | 33 |
| 24 | 39 | 28 |
इसे हल करने के लिए, हम निम्नलिखित चरणों का पालन करेंगे -
- n := पंक्तियों की संख्या, और m = कॉलम की संख्या
- ans नामक एक मैट्रिक्स परिभाषित करें, जिसका ऑर्डर n x m है
- i के लिए 0 से n - 1 तक की रेंज में
- j के लिए 0 से m - 1 तक की रेंज में
- r के लिए i - k से i + k तक की रेंज में
- c के लिए j - k से j + k तक की रेंज में
- यदि r और c मैट्रिक्स इंडेक्स के अंदर हैं, तो
- ans[i, j] := ans[i, j] + mat[r, c]
- यदि r और c मैट्रिक्स इंडेक्स के अंदर हैं, तो
- c के लिए j - k से j + k तक की रेंज में
- r के लिए i - k से i + k तक की रेंज में
- j के लिए 0 से m - 1 तक की रेंज में
- ans लौटाएं
उदाहरण (C++)
बेहतर समझ के लिए निम्नलिखित इम्प्लीमेंटेशन देखें -
#include <bits/stdc++.h> using namespace std; void print_vector(vector<vector<auto> > v){ cout << "["; for(int i = 0; i<v.size(); i++){ cout << "["; for(int j = 0; j <v[i].size(); j++){ cout << v[i][j] << ", "; } cout << "],"; } cout << "]"<<endl; } class Solution { public: vector<vector<int>> matrixBlockSum(vector<vector<int>>& mat, int k) { int n = mat.size(); int m = mat[0].size(); vector < vector <int> > ans(n , vector <int> (m)); for(int i = 0; i < n; i++){ for(int j = 0; j < m; j++){ for(int r = i - k;r <= i + k; r++){ for(int c = j - k; c <= j + k; c++){ if(r>= 0 && r < n && c >= 0 && c < m){ ans[i][j] += mat[r][c]; } } } } } return ans; } }; main(){ vector<vector<int>> v1 = {{1,2,3},{4,5,6},{7,8,9}}; Solution ob; print_vector(ob.matrixBlockSum(v1, 1)); }इनपुट
[[1,2,3],[4,5,6],[7,8,9]] 1
आउटपुट
[[12, 21, 16, ],[27, 45, 33, ],[24, 39, 28, ],]