前缀数组的介绍和使用

我们先来实现一个二维数组的前缀:

int main()

{

int n, m, q;

cin>>n >>m >> q;

vector<vector<int>> arr(n + 1, vector<int>(m + 1));

for (int i = 1; i <= n; i++) {

for (int j = 1; j <= m; j++) {

cin >> arr[i][j];

}

}

vector<vector<long long>> dp(n + 1, vector<long long>(m + 1));

for (int i = 1; i <= n; i++) {

for (int j = 1; j <= m; j++) {

dp[i][j] = dp[i - 1][j] + dp[i][j - 1] + arr[i][j] - dp[i - 1][j - 1];

int x1 = 0, y1 = 0, x2 = 0, y2 = 0;

while (q > 0) {

q--;

cin >> x1 >> y1 >> x2 >> y2;

cout << dp[x2][y2] - dp[x1 - 1][y2] - dp[x2][y1 - 1] + dp[x1 - 1][y1 - 1] << endl;

}

}

}

return 0;

}

class Solution {

public:

vector<int> productExceptSelf(vector<int>& nums) {

int n=nums.size();

vector<int> f(n),g(n);

f[0]=1;

g[n-1]=1;

for(int i=1;i<n;i++){

f[i]=f[i-1]*nums[i-1];

}

for(int i=n-2;i>=0;i--){

g[i]=g[i+1]*nums[i+1];

}

vector<int> ret(n);

for(int i=0;i<n;i++){

ret[i]=f[i]*g[i];

}

return ret;

}

};

class Solution {

public:

int subarraySum(vector<int>& nums, int k) {

unordered_map<int,int> hash;

hash[0]=1;

int sum=0,ret=0;

for(auto e:nums){

sum+=e;

if(hash.count(sum-k)) ret+=hash[sum-k];

hash[sum]++;

}

return ret;

}

};

class Solution {

public:

int pivotIndex(vector<int>& nums) {

int n=nums.size();

vector<int> f(n),g(n);

for(int i=1;i<n;i++){

f[i]=f[i-1]+nums[i-1];

}

for(int j=n-2;j>=0;j--){

g[j]=g[j+1]+nums[j+1];

}

for(int i=0;i<n;i++){

if(f[i]==g[i]){

return i;

}

}

return -1;

}

};

class Solution {

public:

int findMaxLength(vector<int>& nums) {

int n=nums.size();

unordered_map<int,int> hash;

hash[0]=-1;

int sum=0,ret=0;

for(int i=0;i<n;i++){

sum+=nums[i]==0?-1:1;

if(hash.count(sum)) ret=max(ret,i-hash[sum]);

else hash[sum]=i;

}

return ret;

}

};

class Solution {

public:

vector<vector<int>> matrixBlockSum(vector<vector<int>>& mat, int k) {

int m=mat.size(),n=mat[0].size();

vector<vector<int>> dp(m+1,vector<int>(n+1));

for(int i=1;i<=m;i++){

for(int j=1;j<=n;j++){

dp[i][j]=dp[i-1][j]+dp[i][j-1]+mat[i-1][j-1]-dp[i-1][j-1];

}

}

vector<vector<int>> ret(m,vector<int>(n));

for(int i=0;i<m;i++){

for(int j=0;j<n;j++){

int x1=max(0,i-k)+1,y1=max(0,j-k)+1;

int x2=min(m-1,i+k)+1,y2=min(n-1,j+k)+1;

ret[i][j]=dp[x2][y2]-dp[x1-1][y2]-dp[x2][y1-1]+dp[x1-1][y1-1];

}

}

return ret;

}

};

相关推荐
jllllyuz1 天前
MATLAB 蒙特卡洛排队等待模拟程序
数据结构·matlab
自我意识的多元宇宙1 天前
树、森林——树、森林与二叉树的转换(森林转换为二叉树)
数据结构
海清河晏1111 天前
数据结构 | 双循环链表
数据结构·链表
workflower1 天前
机器人应用-楼宇室内巡逻
大数据·人工智能·算法·microsoft·机器人·动态规划·享元模式
ZPC82101 天前
fanuc 机器人通过PR寄存器实现轨迹控制
人工智能·算法·计算机视觉·机器人
py有趣1 天前
力扣热门100题之编辑距离
数据结构·算法·leetcode
Wave8451 天前
C++继承详解
开发语言·c++·算法
Tairitsu_H1 天前
C++类基础概念:定义、实例化和this指针
开发语言·c++
睡觉就不困鸭1 天前
第9天 两数之和
算法·哈希算法·散列表
不想写代码的星星1 天前
C++17 string_view 观察报告:好用,但有点费命
c++