题目链接
超时代码:
class Solution {
public:
/**
* @param heights: a list of integers
* @return: a integer
*/
int trapRainWater(vector<int> &heights) {
// write your code here
int sum=0;
if(!heights.empty())
{
int max=heights[0],temp,i,j,k;//max为向量中的最大值
for(int i=1;i<heights.size();i++)
{
if(heights[i]>max)
{
max=heights[i];
}
}
for(i=0;i<max;i++)
{
for(j=0;j<heights.size();j++)
{
if(heights[j]>0)
{
temp=0;
for(k=j+1;k<heights.size();k++)
{
if(heights[k]<=0)
{
temp++;
}
else
{
sum+=temp;
break;
}
}
}
heights[j]--;
}
}
}
return sum;
}
};
一千多毫秒
主要想法是一层一层的数
优化代码
int trapRainWater(vector<int> &heights) {
// write your code here
int sum=0;
if(!heights.empty())
{
int max=heights[0],maxIndex=0,size=heights.size(),X;//max为向量中的最大值
for(int i=1;i<size;i++)
{
if(heights[i]>max)
{
max=heights[i];
maxIndex = i;
}
}
cout<<max<<" "<<maxIndex<<endl;
X = heights[0];
for(int i=1;i<maxIndex;i++)
{
if(X < heights[i]) X = heights[i];
else sum+=(X-heights[i]);
}
X = heights[size-1];
for(int i=size-2;i>maxIndex;i--)
{
if(X<heights[i]) X=heights[i];
else sum+=(X-heights[i]);
}
}
return sum;
}
才50多毫秒
以前我觉得代码超时是一些需要“优化”而已,但是也许代码超时就是算法错了,优化也许就是换种思路而最优解。
- 超时代码是一层一层的数,LintCode给出数据有3层12个,意味着你至少算3×12次,实际更多
- 优化代码从横向解决问题,它最多只会算12次,所以大大减少了时间,而且代码也不那么杂乱了。
- 以后超时一定多想是不是思路、角度、方法的问题,往往就是这些原因导致的超时