您的位置:首页 > 汽车 > 新车 > leetcode每日一题-3101 交替子数组计数

leetcode每日一题-3101 交替子数组计数

2025/1/11 20:37:54 来源:https://blog.csdn.net/weixin_54423699/article/details/140237239  浏览:    关键词:leetcode每日一题-3101 交替子数组计数

暴力遍历:看起来像是回溯,实际上就是递归

class Solution {
private:long long _res = 0;
public:long long countAlternatingSubarrays(vector<int>& nums) {backtrack(nums, 0);return _res;}void backtrack(vector<int>& nums, long long start){if(start>nums.size())return ;for(long long i=start; i<nums.size(); ++i){// 单个元素必然是交替子数组if(i==start){++_res;}else if(nums[i]!=nums[i-1]){++_res;}else{break;}}backtrack(nums, ++start);}
};

这种方法其实也是可以做的但是会超时哈哈哈。

注意题目给的条件,元素值只有0和1。把握这个点其实就非常好解题了。

假设现在有一个队列0101.....1,队列以元素1结尾说明那么当我们新加入一个元素队列元素总和变了,说明是元素值是1,鉴定重复,元素总和没变说明是0,就是不重复的。队列元素以0结尾情况就恰好相反了。

遍历数组,记录上一个数的数值和当前最长交替子数组的长度。如果当前数和之前的数不一样,则交替子数组的长度加一,否则交替子数组长度为一。
当前最长交替子数组的长度,即是以当前元素为结尾的交替子数组的个数,累加到答案当中。

class Solution {
public:long long countAlternatingSubarrays(vector<int>& nums) {long long res = 0, cur = 0;int pre = -1;for (int a : nums) {cur = (pre != a) ? cur + 1 : 1;pre = a;res += cur;}return res;}
};

=========================================================================

上面是神智不清随便copy的官方题解,下面来说一下这个题目的本质规律。

举个实际的例子,[0, 1, 0, 1, 0]。

设总个数count=0,左边界left从0开始, 右边界right从0开始,那么则有:

left=0, right=0; [0], 满足条件, count+=1, count = 1

left=0, right=1; [0,1], 满足条件, count+=2, count = 3

left=0, right=2; [0,1, 0], 满足条件, count+=3, count = 6

left=0, right=3; [0,1, 0, 1], 满足条件, count+=4, count = 10

left=0, right=4; [0,1, 0, 1, 0], 满足条件, count+=5, count = 10

发现规律了没? count+=(left-right+1)。把握这个规律就可以进行高效的遍历,发现当前位置的元素不符合要求,直接把左边界left挪到当前位置再进行遍历即可。

看到这里如果只追求AC就可以结束了。

再深入的想一下,上述规律为什么是这样子的呢?能不能用数学语言描述一下呢?当然是可以的,聪明的GPT已经给出了答案。

这个题目的本质是在考察,一个长度为n的序列,不重复的子序列有多少?为了节省大家的时间直接截图!

GPT3.5做的,感觉比官方题解好点。

#include <vector>
using namespace std;class Solution {
public:long long countAlternatingSubarrays(vector<int>& nums) {long long count = 0;int left = 0;for (int right = 0; right < nums.size(); ++right) {if (right > 0 && nums[right] == nums[right - 1]) {left = right; // 出现相邻相同元素,更新左边界}count += (right - left + 1); // 加上以 right 为结尾的所有交替子数组数量}return count;}
};

这是一条吃饭博客,由挨踢零声赞助。学C/C++就找挨踢零声,加入挨踢零声,面试不挨踢!

版权声明:

本网仅为发布的内容提供存储空间,不对发表、转载的内容提供任何形式的保证。凡本网注明“来源:XXX网络”的作品,均转载自其它媒体,著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处。

我们尊重并感谢每一位作者,均已注明文章来源和作者。如因作品内容、版权或其它问题,请及时与我们联系,联系邮箱:809451989@qq.com,投稿邮箱:809451989@qq.com