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

相关推荐
Moonbit3 分钟前
月报Vol.03: 新增Bitstring pattern支持,构造器模式匹配增强
后端·算法·github
快手技术8 分钟前
多模态大模型Keye-VL-1.5发布!视频理解能力更强!
算法
薛定谔的算法23 分钟前
JavaScript数组操作完全指南:从基础到高级
前端·javascript·算法
可爱的小小小狼25 分钟前
算法:位运算
算法
VisionPowerful1 小时前
九.弗洛伊德(Floyd)算法
算法·c#
可爱的小小小狼1 小时前
算法:哈希表
redis·算法·散列表
点云侠1 小时前
解决Visual Studio 2022编译工程速度慢的问题
开发语言·c++·ide·算法·计算机视觉·visual studio
THMAIL2 小时前
深度学习从入门到精通 - 迁移学习实战:用预训练模型解决小样本难题
人工智能·python·深度学习·算法·机器学习·迁移学习
.鱼子酱2 小时前
机器学习 - 使用 ID3 算法从原理到实际举例理解决策树
算法·决策树·机器学习
Swift社区2 小时前
Swift 解法详解:LeetCode 371《两整数之和》
开发语言·leetcode·swift