题目描述
桌子上垂直堆叠着 N 个立方体。
你得到一个长度为 N 的字符串 S。如果 S 中的第 i 个字符为 0,则从底部开始的第 i 个立方体的颜色为红色,如果该字符为 1,则从底部开始的第 i 个立方体的颜色为蓝色。
您可以多次执行以下操作:选择相邻的红色立方体和蓝色立方体,然后将其删除。在这里,堆叠在移除的立方体上的立方体将落到它们下面的对象上。
最多可以移除多少个立方体?
约束
·1≤N≤105
·|S|=N
·S 中的每个字符都是 0 或 1。
你得到一个长度为 N 的字符串 S。如果 S 中的第 i 个字符为 0,则从底部开始的第 i 个立方体的颜色为红色,如果该字符为 1,则从底部开始的第 i 个立方体的颜色为蓝色。
您可以多次执行以下操作:选择相邻的红色立方体和蓝色立方体,然后将其删除。在这里,堆叠在移除的立方体上的立方体将落到它们下面的对象上。
最多可以移除多少个立方体?
约束
·1≤N≤105
·|S|=N
·S 中的每个字符都是 0 或 1。
输入
输入由标准输入以以下格式给出:
S
S
输出
打印可删除的最大立方体数。
样例输入 Copy
0011
样例输出 Copy
4
提示
可以通过执行以下操作来删除所有四个立方体:
·从底部取出第二个和第三个立方体。然后,第四个立方体落到第一个立方体上。
·从底部取出第一个和第二个立方体。
·从底部取出第二个和第三个立方体。然后,第四个立方体落到第一个立方体上。
·从底部取出第一个和第二个立方体。
#include <bits/stdc++.h>
using namespace std;
stack<char> s;//建立一个char类型的栈,这里我原来是string类型的,但是不知道为什么报错,求大佬指正
int main()
{
string a;
int n,i,k=0;
cin>>a;
for(i=0;i<a.size();i++)
{
if(s.empty())//如果这个栈是空的,那就直接将当前元素压入栈顶
s.push(a[i]);
else if(a[i]!=s.top())//如果当前元素不等于栈顶元素
{
s.pop();//那就说明这两个立方体的颜色不同,就删除当前栈顶的内容
k+=2;//并且让计数器+2,因为有两个元素符合题意了
}
else if(a[i]==s.top())//如果这两个元素相等的话,就将当前这个元素压入栈顶
s.push(a[i]);
}
cout<<a.size()-s.size();//用原来字符串的长度减去栈中数据的长度就是答案,或者直接输出k
}