力扣9.27

2516. 每种字符至少取 K 个

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

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

数据范围

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

分析

本题有良好的二分性质,对于某一个 t t t,小于 t t t的分钟都无法满足,大于等于t的都可以满足,因此可以考虑二分答案

代码

c 复制代码
class Solution {
public:
    const static int N = 1e5 + 5;
    int prea[N], preb[N], prec[N];
    int lasta[N], lastb[N], lastc[N];
    int n;
    bool check(int t, int k) {
        for(int i = 0; i <= t; i ++ ) {
            int na = prea[i] + lasta[n - (t - i) + 1];
            int nb = preb[i] + lastb[n - (t - i) + 1];
            int nc = prec[i] + lastc[n - (t - i) + 1];
            if(na >= k && nb >= k && nc >= k) return true;
        }
        return false;
    }
    int find(int l, int r, int k) {
        while(l < r) {
            int mid = (l + r) >> 1;
            if(check(mid, k)) r = mid;
            else l = mid + 1;
        }
        return l;
    }
    int takeCharacters(string s, int k) {
        n = s.size();
        for(int i = 1; i <= n; i ++ ) {
            prea[i] = prea[i - 1] + (s[i  - 1] == 'a' ? 1 : 0);
            preb[i] = preb[i - 1] + (s[i  - 1] == 'b' ? 1 : 0);
            prec[i] = prec[i - 1] + (s[i  - 1] == 'c' ? 1 : 0);
        }
        for(int i = n; i >= 1; i -- ) {
            lasta[i] = lasta[i + 1] + (s[i - 1] == 'a' ? 1 : 0);
            lastb[i] = lastb[i + 1] + (s[i - 1] == 'b' ? 1 : 0);
            lastc[i] = lastc[i + 1] + (s[i - 1] == 'c' ? 1 : 0);
        }
        if(prea[n] < k || preb[n] < k || prec[n] < k) return -1;
        int l = 0, r = n;
        int res = find(0, n, k);
        return res;
    }
};

相关推荐
不知天地为何吴女士1 小时前
Day32| 509. 斐波那契数、70. 爬楼梯、746. 使用最小花费爬楼梯
算法
小坏坏的大世界1 小时前
C++ STL常用容器总结(vector, deque, list, map, set)
c++·算法
励志要当大牛的小白菜4 小时前
ART配对软件使用
开发语言·c++·qt·算法
qq_513970444 小时前
力扣 hot100 Day56
算法·leetcode
PAK向日葵5 小时前
【算法导论】如何攻克一道Hard难度的LeetCode题?以「寻找两个正序数组的中位数」为例
c++·算法·面试
爱喝矿泉水的猛男7 小时前
非定长滑动窗口(持续更新)
算法·leetcode·职场和发展
YuTaoShao7 小时前
【LeetCode 热题 100】131. 分割回文串——回溯
java·算法·leetcode·深度优先
YouQian7727 小时前
Traffic Lights set的使用
算法
go54631584659 小时前
基于深度学习的食管癌右喉返神经旁淋巴结预测系统研究
图像处理·人工智能·深度学习·神经网络·算法
aramae9 小时前
大话数据结构之<队列>
c语言·开发语言·数据结构·算法