气温变化趋势

2024-06-21

题目:

力扣城计划在两地设立「力扣嘉年华」的分会场,气象小组正在分析两地区的气温变化趋势,对于第 i ~ (i+1) 天的气温变化趋势,将根据以下规则判断:

  • 若第 i+1 天的气温 高于i 天,为 上升 趋势
  • 若第 i+1 天的气温 等于i 天,为 平稳 趋势
  • 若第 i+1 天的气温 低于i 天,为 下降 趋势

已知 temperatureA[i]temperatureB[i] 分别表示第 i 天两地区的气温。 组委会希望找到一段天数尽可能多,且两地气温变化趋势相同的时间举办嘉年华活动。请分析并返回两地气温变化趋势相同的最大连续天数

即最大的 n,使得第 i~i+n 天之间,两地气温变化趋势相同

示例 1:

输入: temperatureA = [21,18,18,18,31] temperatureB = [34,32,16,16,17]

输出:2

解释:如下表所示, 第 2~4 天两地气温变化趋势相同,且持续时间最长,因此返回 4-2=2image.png

示例 2:

输入: temperatureA = [5,10,16,-6,15,11,3] temperatureB = [16,22,23,23,25,3,-16]

输出:3

提示:

  • 2 <= temperatureA.length == temperatureB.length <= 1000
  • -20 <= temperatureA[i], temperatureB[i] <= 40

思路:

我们关心的只有气温的变化趋势,至于气温是多少那无所谓

所以我们先将气温的变化趋势求出来,上升记为1,平稳记为0,下降记为-1
然后我们依次比较A,B的变化趋势,求出最大相同的子数组即可

代码:

class Solution {
public:
    int temperatureTrend(vector<int>& temperatureA, vector<int>& temperatureB) {
        int n = temperatureA.size();
        vector<int> change_A(n - 1);
        vector<int> change_B(n - 1);

        for (int i = 0; i < n - 1; i++) {
            if (temperatureA[i] < temperatureA[i + 1]) change_A[i] = 1;
            else if (temperatureA[i] == temperatureA[i + 1]) change_A[i] = 0;
            else change_A[i] = -1;

            if (temperatureB[i] < temperatureB[i + 1]) change_B[i] = 1;
            else if (temperatureB[i] == temperatureB[i + 1]) change_B[i] = 0;
            else change_B[i] = -1;
        }

        int ans = 0;
        for (int i = 0; i < n - 1; i++) {
            if (change_A[i] == change_B[i]) {
                int tmp = i;
                while (i < n - 1 && change_A[i] == change_B[i]) i++;
                ans = max(ans, i - tmp);
                i--;
            }
        }
        return ans;
    }
};