每日一题 day 40 (DP topic)

最新推荐文章于 2024-11-04 19:24:37 发布

NP_hard

最新推荐文章于 2024-11-04 19:24:37 发布

阅读量178

点赞数

分类专栏： LeetCode Learning 文章标签：动态规划算法 leetcode

本文链接：https://blog.csdn.net/np_hard/article/details/123251011

版权

LeetCode Learning 专栏收录该内容

60 篇文章 0 订阅

订阅专栏

这篇博客探讨了如何解决LeetCode的413题——等差数列子序列。提供了三种不同的解决方案：暴力求解、不使用额外空间的方法以及最短代码实现。每种方法都详细解释了其工作原理，并通过示例输入和输出展示了它们的正确性。这些算法主要涉及数组处理和动态规划思想。

摘要由CSDN通过智能技术生成

problem

413. Arithmetic Slices
An integer array is called arithmetic if it consists of at least three elements and if the difference between any two consecutive elements is the same.

For example, [1,3,5,7,9], [7,7,7,7], and [3,-1,-5,-9] are arithmetic sequences.
Given an integer array nums, return the number of arithmetic subarrays of nums.

A subarray is a contiguous subsequence of the array.

Example 1:

Input: nums = [1,2,3,4]
Output: 3
Explanation: We have 3 arithmetic slices in nums: [1, 2, 3], [2, 3, 4] and [1,2,3,4] itself.

Example 2:

Input: nums = [1]
Output: 0

approach 1 `brute force`

class Solution {
public:
    int numberOfArithmeticSlices(vector<int>& nums) {
        if(nums.size() < 3)return 0;
        vector<int> diff(nums.size());
        diff[0] = -114514;
        for(int i=1; i<nums.size(); i++)
            diff[i] = nums[i]-nums[i-1];
        int res = 0, i=2;
        while(i < nums.size()){
            int cnt = 1;
            while(i < nums.size() && diff[i-1]==diff[i])
                ++cnt, ++i;
            if(cnt > 1) res += (cnt*(cnt-1))/2;
            else ++i;
        }
        return res;
    }
};

在这里插入图片描述

approach 2 `without extra space`

class Solution {
public:
    int numberOfArithmeticSlices(vector<int>& nums) {
        if(nums.size() < 3)return 0;
        int res = 0, cnt = 1;
        for(int i=2; i<nums.size(); i++){
            if(nums[i]+nums[i-2] == 2*nums[i-1]){
                ++cnt;
            }else{
                res += (cnt*(cnt-1))/2;
                cnt = 1;
            }
        }
        if(cnt > 1) res +=(cnt*(cnt-1))/2;
        return res;
    }
};

在这里插入图片描述

approach 3 `shortest code, but not best`

class Solution {
public:
    int numberOfArithmeticSlices(vector<int>& nums) {
        if(nums.size() < 3)return 0;
        int res = 0;
        for(int i=2, j=1; i<nums.size(); i++){
            if(nums[i]+nums[i-2] != 2*nums[i-1])
                j=i;
            res += i-j;
        }
        return res;
    }
};