刷题的日常-分割字符串的最大得分

71 阅读2分钟

携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第19天,点击查看活动详情

刷题的日常-2022年8月14日

一天一题,保持脑子清爽

分割字符串的最大得分

来自leetcode的 1422 题,题意如下:

给你一个由若干 0 和 1 组成的字符串 s ,请你计算并返回将该字符串分割成两个 非空 子字符串(即左 子字符串和 右 子字符串)所能获得的最大得分。
「分割字符串的得分」为 左 子字符串中 0 的数量加上 右 子字符串中 1 的数量。

示例如下:

输入:s = "011101"
输出:5 
解释:
将字符串 s 划分为两个非空子字符串的可行方案有:
左子字符串 = "0" 且 右子字符串 = "11101",得分 = 1 + 4 = 5 
左子字符串 = "01" 且 右子字符串 = "1101",得分 = 1 + 3 = 4 
左子字符串 = "011" 且 右子字符串 = "101",得分 = 1 + 2 = 3 
左子字符串 = "0111" 且 右子字符串 = "01",得分 = 1 + 1 = 2 
左子字符串 = "01110" 且 右子字符串 = "1",得分 = 2 + 1 = 3

理解题意

我们可以从题意中提取的条件如下:

  • 给定一个由0 和 1组成的字符串
  • 对其进行划分,要求左边统计0的个数,右边统计1的个数
  • 将左右的和最大值返回

做题思路

我们可以先统计1的个数,因为1是在右边的,统计完成之后再进行划分:

  • 统计字符串中1出现的次数
  • 先划分第一个数,去除边界
  • 然后往后遍历,如果出现1的话就减去1的个数
  • 出现0则加上0的个数
  • 划分的过程中要判断是否是最大划分
  • 最后返回最大划分

代码实现

代码实现如下:因为需要遍历两次字符串,所以时间复杂度为O(s.length())

public class Solution {
    public int maxScore(String s) {
        int o = 0, z = 0;
        for (int i = 0; i < s.length(); i++) {
            char c = s.charAt(i);
            o += c == '1' ? 1 : 0;
        }
        char c = s.charAt(0);
        o -= c == '1' ? 1 : 0;
        z += c == '0' ? 1 : 0;
        int result = o + z;
        for (int i = 1; i < s.length() - 1; i++) {
            char ch = s.charAt(i);
            o -= ch == '1' ? 1 : 0;
            z += ch == '0' ? 1 : 0;
            result = Math.max(result, o + z);
        }
        return result;
    }
}