力扣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;
    }
};

相关推荐
西阳未落30 分钟前
LeetCode——双指针(进阶)
c++·算法·leetcode
一二学长44 分钟前
异或相关的算法题
算法
暴力求解1 小时前
c++类和对象(下)
开发语言·c++·算法
深栈2 小时前
机器学习:支持向量机
算法·机器学习·支持向量机
刘海东刘海东2 小时前
结构型智能科技理论研究(草稿)
科技·算法
C嘎嘎嵌入式开发2 小时前
(10)100天python从入门到拿捏《Python中的数据结构与自定义数据结构》
数据结构·python·算法
熬了夜的程序员2 小时前
【LeetCode】69. x 的平方根
开发语言·算法·leetcode·职场和发展·动态规划
Niuguangshuo2 小时前
音频特征提取算法介绍
算法·音视频
fengfuyao9852 小时前
基于MATLAB的匈牙利算法实现任务分配
算法·数学建模·matlab
CoovallyAIHub3 小时前
超详细链式插补 (MICE) 多元插补:机器学习模型的高级缺失数据处理
算法·机器学习·计算机视觉