力扣:1422. 分割字符串的最大得分

243 阅读3分钟

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

1422. 分割字符串的最大得分

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

「分割字符串的得分」为 左 子字符串中 0 的数量加上 右 子字符串中 1 的数量。

示例 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

示例 2:
输入:s = "00111"
输出:5
解释:当 左子字符串 = "00" 且 右子字符串 = "111" 时,我们得到最大得分 = 2 + 3 = 5

示例 3:
输入:s = "1111"
输出:3

提示:

  • 2 <= s.length <= 500
  • 字符串 s 仅由字符 '0' 和 '1' 组成。

问题解析

暴力做法: 因为数据很小,所以可以直接从1到n(字符串长度)枚举分割点i,把字符串分成左右两份:

  • 左边的字符串,下标0到下标i-1,得分为0的个数。
  • 右边的字符串,下标i到下标n-1,得分为1的个数。

然后再遍历两个字符串求的他们的分数,在此过程中维护最大值即可。

class Solution {
public:
    int getnum(string s,char c)
    {
        int num=0;
        for(auto i:s)
        {
            if(i==c)num++;
        }
        return num;
    }
    int maxScore(string s) {
        int n=s.size(),mx=0;
        for(int i=1;i<n;i++)
        {
            string a=s.substr(0,i),b=s.substr(i,n-i);
            int x=getnum(a,'0')+getnum(b,'1');
            mx=max(mx,x);
        }
        return mx;
    }
};

优化: 因为每次分割点右移,改变的只有分割点上的那个字符而已:上个分割点的字符分给了左边,右边少了这个字符。 那么只要统计一下一开始的分数,然后每次根据分割点的字符来判断分数修改即可:

  • 如果当前字符是‘1’,那么右边字符串分数-1,左边分数不变
  • 如果当前字符是‘0’,那么左边字符串分数+1,右边分数不变
class Solution {
public:
    int getnum(string s,char c)
    {
        int num=0;
        for(auto i:s)
        {
            if(i==c)num++;
        }
        return num;
    }
    int maxScore(string s) {
        int n=s.size(),mx=0;
        string a=s.substr(0,1),b=s.substr(1,n-1);
        int x=getnum(a,'0'),y=getnum(b,'1');
        mx=max(mx,x+y);
        for(int i=1;i<n-1;i++)
        {
            if(s[i]=='1')
            {
                y--;
            }
            else x++;
            
            mx=max(mx,x+y);
        }
        return mx;
    }
};