leetcode327

来源:互联网 发布:凌志软件 编辑:程序博客网 时间:2024/06/03 20:21
  • 超时
    该题目一开始想采用DP的思想,从前往后统计每一个位置可能的取值情况。
Data MightValue 4 4 2 2,6 -1 -1,1,5

每一个值都有自己可能的取值。但是超时了,vector的更新还是太会费时间。

注意:要用double类型,或者long long类型,因为不同的int类型相加会出现内存泄漏的情况

class Solution {public:    vector<double> newList(int number, vector<double> oldList,int& count,int lower,int upper)    {        vector<double> result;        result.push_back(number);        if(number<=(double)upper && number>=(double)lower)            count++;        for(int i=0;i<oldList.size();i++)        {            result.push_back(double(number)+double(oldList[i]));            if(result.back()<=(double)upper && result.back()>=(double)lower)                count++;        }        return result;    }    int countRangeSum(vector<int>& nums, int lower, int upper) {        if(nums.size()==0)            return 0;        int count=0;        vector<double> list;        for(int i=0;i<nums.size();i++)            list=newList(nums[i],list,count,lower,upper);        return count;    }};
  • 改进代码(两种方法是一样的,只不过这种方法避免了vector的不断更新,而是保存了其sum)
    因此采用另一种方法,这种方法利用multiset类型。
    值得注意的是: multiset可以维持一个有元素重复的数组,并且数组是按照从小到大的顺序自动存储。set则不允许有元素重复。map不允许key重复,不过也是安装key值自动从小到大排序。

正确代码采用了另一种处理思路。最主要的是不再处理原始数据,而是处理他们的sum。由于计算的区间是连续区间,所以[i,j] 区间的和就是sum[j]-sum[i]。其中有几个特殊的地方需要注意:

  1. 当i=0时,区间[0,j]的和就是sum[j]-sum[0]=sum[j],因此sum[0]=0。所以一开始,数组需要先增加一个元素0。
  2. 当i=j时,区间[j,j]的和就是sum[j]-sum[j]=0。不对,需要单独处理。
class Solution {public:    int countRangeSum(vector<int>& nums, int lower, int upper) {        if(nums.size()==0)            return 0;        int count=0;        multiset<double> sumSet;        double sum=0;        sumSet.insert(0);        for(int i=0;i<nums.size();i++)        {            sum+=(double)nums[i];            multiset<double>::iterator itStart,itEnd;            itStart=sumSet.lower_bound(sum-(double)upper);            itEnd=sumSet.upper_bound (sum-(double)lower);            count+=std::distance(itStart,itEnd);            cout<<count<<" ";            sumSet.insert(sum);        }        return count;    }};
0 0
原创粉丝点击