区域和检索——数组不可变

前两天终于实现了翻墙,现在开刷leetcode,久闻大名,今天才开始真是惭愧!

刷题的奥义在于一定要分tag,一段时间(比如一周)只刷一种能行,浅尝辄止随便去刷很容易忘记,单恋一种题型又容易被困住思维,本周我选择动态规划!刷个十道,同时C++和java混用着做


给定一个整数数组 nums,求出数组从索引 i 到 j (i ≤ j) 范围内元素的总和,包含 i, j 两点。

示例:

给定 nums = [-2, 0, 3, -5, 2, -1],求和函数为 sumRange()

sumRange(0, 2) -> 1
sumRange(2, 5) -> -1
sumRange(0, 5) -> -3
说明:

你可以假设数组不可变。
会多次调用 sumRange 方法。


其实这道题不难,一眼就看出用数组循环

class NumArray {
public:
    vector<int> nums;
    NumArray(vector<int> nums) {
        this->nums=nums;
    }
    
    int sumRange(int i, int j) {
        int sum=0;
        for(int b=i;b<j+1;b++){
            sum+=nums[b];
        }
        return sum;
    }
};

然而显示我速度只超过了2%的人,为什么不贴java的,因为java直接超时,过不了!

上面的代码的问题在于,当测试用例过多时,每次都必须重新计算,但实际上,由于数组没变,所以可以事先计算好大部分数据,用另外一个数组来保存,而此时,就用动态规划来保存

class NumArray {
    int [] nums;
    int len;
    int [] sum;
    public NumArray(int[] nums) {
        this.nums=nums;
        len=nums.length;
        if(len>0){
            sum=new int [len+1]
            for(int i=1;i<len+1;i++)
            //sum[i]不包括nums[i],因为sumRange中会被减掉
            sum[i]=sum[i-1]+nums[i-1];
        }
    }
    public int sumRange(int i, int j) {
        int s=0;
        s=sum[j+1]-sum[i];
        return s;
    }
}

上面的代码不仅可以用java,而且超过了80%的人哦!

那你可能会好奇,最快的代码是什么样的呢?
下面这个,同样的动态规划,用时只有我的一半!

class NumArray {

    int[] nums;
    public NumArray(int[] nums) {
    //这里全盘使用this指针,创建的变量数达到最少
        this.nums = new int[nums.length];
        if(this.nums.length == 0) return;
        this.nums[0] = nums[0];
        for(int i=1; i<nums.length;i++) {
            this.nums[i] = this.nums[i-1] + nums[i];
        }
    }

    public int sumRange(int i, int j) {
    //这里他没有新设变量,而是直接返回结果
        return i==0? nums[j]:nums[j]-nums[i-1];
    }
}
全部评论

相关推荐

服从性笔试吗,发这么多笔,现在还在发。
蟑螂恶霸zZ:傻 x 公司,发两次笔试,两次部门匹配挂,
投递金山WPS等公司10个岗位 >
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务