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

前言

小白算法比较菜，希望能激励我每日更新，从leetcode第一题开始，2022年目标2000分，现在1968！！

力扣第 325 场周赛-力扣

第 325 场周赛

阳了，没起来参加

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的个数，一旦满足就返回true即可。

方法

class Solution {
    public int takeCharacters(String s, int k) {
        if(k == 0) {
            return 0;
        }
        int a = 0;
        int b = 0;
        int c = 0;
        char[] chars = s.toCharArray();
        for(char ch : chars) {
            if(ch == 'a') {
                a++;
            } else if(ch == 'b') {
                b++;
            } else {
                c++;
            }
        }
        if(c < k || b < k || a < k) {
            return -1;
        }
        int l = 3 * k;
        int r = chars.length;
        while(l < r) {
            int mid = l + r >> 1;
            if(fun(mid, chars, k)) {
                r = mid;
            } else {
                l = mid + 1;
            }
        }
        return l;
    }
    public boolean fun(int mid, char[] chars, int k) {
        int a = 0;
        int b = 0;
        int c = 0;
        for(int i = 0; i < mid; i++) {
            if(chars[i] == 'a') {
                a++;
            } else if(chars[i] == 'b') {
                b++;
            } else {
                c++;
            }
        }
        if(a >= k && b >= k && c >= k) {
            return true;
        }
        for(int i = 0; i < mid; i++) {
            int index = mid - i - 1;
            if(chars[index] == 'a') {
                a--;
            } else if(chars[index] == 'b') {
                b--;
            } else {
                c--;
            }
            index = chars.length - 1 - i;
            if(chars[index] == 'a') {
                a++;
            } else if(chars[index] == 'b') {
                b++;
            } else {
                c++;
            }
            if(a >= k && b >= k && c >= k) {
                return true;
            }
        }
        return false;
    }
}