首页 > 技术文章 > leetcode 1588 int运算时的优化导致实现有误问题

fdbwymz 2021-08-30 11:02 原文

1588. 所有奇数长度子数组的和 - 力扣(LeetCode) (leetcode-cn.com)

1.算法

     暴力解法 复杂度 O(n^3)

     前缀和优化  复杂度O (n^2);

     数学方法     复杂度O(n);

     数学方法公式推导: 

     对于下标为 i 的 项, 包含i 项的子数组有以下情况:

         1. 前边偶数个+ 1                                [ i / 2 ]   (这个符号说明向下取整)

         2.右边偶数个+1                                 [ (n-i-1)/2 ]  

         3 .左边偶数个+1+右边偶数个            [ (i+1)/2 ] * [  (n-i) /2 ]

         4.左边奇数个+1+ 右边奇数个            [ i/2 ]  * [ (n-i-1)/2 ]           

         5. 它自身                                             1

2. 出现向下取整运算的注意!!

      对于上边总结的五种情况 如果使用代码: 

      sum=i/2+(n-i-1)/2 +(i+1)/2*(n-i)/2 +(i/2)*(n-i-1)/2 +1; 

     就会出现错误,可能因为编译器的优化会将一些项合并,取整的含义被优化消失了

     因此应该将每一项单独运算再相加,或者用int () 将每一次向上取整项括起来。

3.代码

    

class Solution {
public:
    int sumOddLengthSubarrays(vector<int>& arr) {
        //数学
        int ret=0;
        int n= arr.size();
        for(int i=0;i<n;i++){
            int sum=0;
            if(i>0&&n-i-1>0){
                sum=int(i/2)+int((n-i-1)/2)+int((i+1)/2)*int((n-i)/2)+
                int(i/2)*int((n-i-1)/2)+1;
            }
            if(i==0&&n-i-1>0){
                sum=(n-i-1)/2+1;
            }
            if(i>0&&n-i-1==0)
                sum=i/2+1;
            if(i==0&&n-i-1==0)
                sum=1;
            ret+=sum*arr[i];
        }
        return ret;
    }
};

 

     

     

推荐阅读