桌子上有 n
个球,每个球的颜色不是黑色,就是白色。
给你一个长度为 n
、下标从 0 开始的二进制字符串 s
,其中 1
和 0
分别代表黑色和白色的球。
在每一步中,你可以选择两个相邻的球并交换它们。
返回「将所有黑色球都移到右侧,所有白色球都移到左侧所需的 最小步数」。
示例 1:
输入:s = "101"
输出:1
解释:我们可以按以下方式将所有黑色球移到右侧:
- 交换 s[0] 和 s[1],s = "011"。
最开始,1 没有都在右侧,需要至少 1 步将其移到右侧。
示例 2:
输入:s = "100"
输出:2
解释:我们可以按以下方式将所有黑色球移到右侧:
- 交换 s[0] 和 s[1],s = "010"。
- 交换 s[1] 和 s[2],s = "001"。
可以证明所需的最小步数为 2 。
示例 3:
输入:s = "0111"
输出:0
解释:所有黑色球都已经在右侧。
提示:
·1 <= n == s.length <= 105
·s[i]
不是 '0'
,就是 '1'
。
题目大意:在每次操作只能交换相邻字符的情况下计算将0全都移到最左边需交换的最少次数。
分析:设字符串长度为N
(1)将所有0移到最左边等同于将所有1移到最右边;
(2)原字符串中最右边的1移动到最终字符串的最后1个字符时交换次数最少,同理原字符串中倒数第2个1移动到最终字符串的倒数第2个字符时交换次数最少;
(3)由(2)可得,原字符串中倒数第k个1移动到最终字符串的倒数第k个字符时交换次数最少,因此设原字符串中倒数第k个1的下标为i(从0开始),则这个1的交换次数为N-k-i时,总的交换次数最少。
class Solution {
public:
long long minimumSteps(string s) {
long long ans=0;
for(int N=s.size(),i=N-1,count=0;i>=0;--i){
if(s[i]=='1') ans+=N-(++count)-i;
}
return ans;
}
};