leetcode327
来源:互联网 发布:凌志软件 编辑:程序博客网 时间:2024/06/03 20:21
- 超时
该题目一开始想采用DP的思想,从前往后统计每一个位置可能的取值情况。
每一个值都有自己可能的取值。但是超时了,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]。其中有几个特殊的地方需要注意:
- 当i=0时,区间[0,j]的和就是sum[j]-sum[0]=sum[j],因此sum[0]=0。所以一开始,数组需要先增加一个元素0。
- 当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