- 超时
该题目一开始想采用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]。其中有几个特殊的地方需要注意:
- 当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;
}
};