LeetCode 2516.每种字符至少取k个(c++)

252 阅读1分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第31天,点击查看活动详情

前言

从零开始学习c++,每天起码做一道leetcode题目,在此记录,希望最后能够有所收获!

一、题目描述

2516.每种字符至少取k个

给你一个由字符 'a'、'b'、'c' 组成的字符串 s 和一个非负整数 k 。每分钟,你可以选择取走 s 最左侧 还是 最右侧 的那个字符。

你必须取走每种字符 至少 k 个,返回需要的 最少 分钟数;如果无法取到,则返回 -1 。

  示例 1:

输入:s = "aabaaaacaabc", k = 2
输出:8
解释:
从 s 的左侧取三个字符,现在共取到两个字符 'a' 、一个字符 'b' 。
从 s 的右侧取五个字符,现在共取到四个字符 'a' 、两个字符 'b' 和两个字符 'c' 。
共需要 3 + 5 = 8 分钟。
可以证明需要的最少分钟数是 8

示例 2:

输入: s = "a", k = 1
输出: -1
解释: 无法取到一个字符 'b' 或者 'c',所以返回 -1

提示:

  • 1 <= s.length <= 105
  • s 仅由字母 'a''b''c' 组成
  • 0 <= k <= s.length

二、思路分析

这题感觉是比较有难度的,因为是从左右两边来进行选择的,这给我一种动态和求最优解的的感觉,在每次选择时应该选择左边还是右边呢?

后来看了别人的做法,主要分为两种思路,一种是二分(二分永远滴神!),但挺难理解的;另一种是是滑动窗口,我觉得这种对于我来说更容易理解。

滑动窗口的主要思路是将问题的求解转化为:从中间去掉一段连续的序列,使得剩下的元素中a、b、c每个字母的数量均大于等于k。这样一来就比较清晰了。j在前,i在后,作为一个滑动的窗口。

三、AC代码

class Solution {
public:
    int takeCharacters(string s, int k) {
        int a,b,c,n=s.size();
        a=count(s.begin(),s.end(),'a');
        b=count(s.begin(),s.end(),'b');
        c=count(s.begin(),s.end(),'c');
        if(a<k||b<k||c<k)
            return -1;
        a=a-k,b=b-k,c=c-k;
        vector<int>num(3),check{a,b,c};
        int ans=INT_MAX,i,j;
        for(i=0,j=0;j<s.size();j++)
        {
            num[s[j]-'a']++;
            if(num[s[j]-'a']>check[s[j]-'a'])
            {
                while(num[s[j]-'a']>check[s[j]-'a'])
                {
                    num[s[i]-'a']--;
                    i++;
                }
            }
            ans=min(n-(j-i+1),ans);
        }
        return ans;
    }
};

四、总结

又见了一种题型。