2024年4月19號題解

560.和為k的子陣列

題目描述

給你一個整數陣列 nums 和一個整數 k ，請你統計並返回 該陣列中和為 k 的子陣列的個數 。

子陣列是陣列中元素的連續非空序列。

示例 1：

輸入：nums = [1,1,1], k = 2
輸出：2

示例 2：

輸入：nums = [1,2,3], k = 3
輸出：2

提示：

• 1 <= nums.length <= 2 * 104
• -1000 <= nums[i] <= 1000
• -107 <= k <= 107

解題思路

1. 遍歷所有子陣列的左邊界和右邊界
2. 求和區間中的和並判斷等不等於k

程式碼實現

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        int ans = 0;

        for (int i = 0; i < nums.size(); i ++) {
            int sum = nums[i];
            if (sum == k) ans ++;
            for (int j = i + 1; j < nums.size(); j ++) {
                sum += nums[j];
                if (sum == k) ans ++;
            }
        }

        return ans;
    }
};

128.最長連續序列

題目描述

給定一個未排序的整數陣列 nums ，找出數字連續的最長序列（不要求序列元素在原陣列中連續）的長度。

請你設計並實現時間複雜度為 O(n) 的演算法解決此問題。

示例 1：

輸入：nums = [100,4,200,1,3,2]
輸出：4
解釋：最長數字連續序列是 [1, 2, 3, 4]。它的長度為 4。

示例 2：

輸入：nums = [0,3,7,2,5,8,4,6,0,1]
輸出：9

提示：

• 0 <= nums.length <= 105
• -109 <= nums[i] <= 109

解題思路

1. 給陣列排序
2. 遍歷陣列，並判斷是否連續
3. 不連續就更新答案，連續就長度加一

程式碼實現

class Solution {
public:
    int longestConsecutive(vector<int>& nums) {    
        int ans = 0;
        int len = 1;
        set<int> s;

        for (auto it = nums.begin(); it != nums.end(); it ++) {
            s.insert(*it);
        }

        for (auto it = s.begin(); it != s.end(); it ++) {
            if ((next(it) != s.end()) && *next(it) - *it == 1) {
                len ++;
            }
            else {
                ans = max(ans, len);
                len = 1;
            }
        }

        return ans;
    }
};