Hugo Future Imperfect Slim

邱圆辉

未来可期

前缀和模板

一维前缀和以及二维前缀和的代码模板

邱圆辉

1 分钟

一维前缀和

vector<int> one_dimension_prefix(vector<int> nums) {
    int n = nums.size();
	vector<int> prefix(n + 1);
    for (int i = 1; i <= n; i++) {
        prefix[i] = prefix[i - 1] + nums[i - 1];
    }
    return prefix;
}

二维前缀和

vector<vector<int>> two_dimension_prefix(vector<vector<int>> nums) {
    int m = nums.size();
    int n = nums[0].size();
    vector<vector<int>> prefix(m + 1, vector(n + 1, 0));
    for (int i = 1; i <= m; i++) {
        for (int j = 1; j <= n; j++) {
            prefix[i][j] = nums[i][j] + prefix[i - 1][j]
                + prefix[i][j - 1] - prefix[i - 1][j - 1];
        }
    }
    return prefix;
}

说些什么

评论

还沒有留言。

最新文章

分类

关于

This theme was developed for Hugo static site generator.