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