题目:
力扣城计划在两地设立「力扣嘉年华」的分会场,气象小组正在分析两地区的气温变化趋势,对于第 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=2
示例 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;
    }
};
