题目链接: C. Binary String
题意: 给一个01字符串,从字符串两侧删若干个数,对剩下的0和删掉的1取最大值,让这个最大值最小
思路: 前缀和, 我们注意到如果剩下的0比删掉的1多很多的话,显然不是最优解,故我们先统计下0的总数 cnt
先从左往右遍历,统计0的前缀和 pre0[i] 与1的前缀和 pre1[i] 以及两者后缀和 hou0[i] , hou1[i] ,这样对于每个点,就能得到剩下的0的数量 cnt-pre0[i] 和删掉1的数量 pre1[i] ,取其差值
res=cnt-pre0[i]-pre1[i]
然后考虑从右边删,容易想到当删掉的1的个数比剩下的0的个数还要多的话,继续删就没意义了,因为这样最后值只会越来越大,所以只统计 res>=0 的情况
我们右边删的时候让剩下的0(res1)和删掉1 (res2) 相等,而不管删的是0还是1,都会让 res1 和 res2 更加接近,删掉的值刚好就是 res ,即前面的差值,左右都删了,对剩下的0的个数
ans=cnt-pre0[i]-hou0[pos1]
取min,然后再从右往左镜像操作一遍,就能得到答案
Code:
#include <bits/stdc++.h>
#define debug freopen("_in.txt", "r", stdin);
typedef long long ll;
typedef unsigned long long ull;
typedef struct Node *bintree;
using namespace std;
const ll INF = 0x3f3f3f3f3f3f3f3f;
const ll maxn = 1e6 + 10;
const ll maxm = 1e5 + 10;
const ll mod = 1e9 + 7;
const double pi = acos(-1);
const double eps = 1e-8;
#define lson l, m, rt << 1
#define rson m + 1, r, rt << 1 | 1
ll T;
ll n, m, maxx;
char str[maxn];
ll pre0[maxn], pre1[maxn], hou0[maxn], hou1[maxn];
void solve()
{
scanf("%s", str + 1);
ll len = strlen(str + 1);
n = len;
ll cnt = 0;
for (ll i = 1; i <= n; i++)
{
if (str[i] == '0')
cnt++;
}
for (ll i = 1; i <= n; i++)
{
if (str[i] == '0')
{
pre0[i] = pre0[i - 1] + 1;
pre1[i] = pre1[i - 1];
}
else
{
pre0[i] = pre0[i - 1];
pre1[i] = pre1[i - 1] + 1;
}
}
hou0[n+1]=hou1[n+1]=0;
for (ll i = n; i >= 1; i--)
{
if (str[i] == '0')
{
hou0[i] = hou0[i + 1] + 1;
hou1[i] = hou1[i + 1];
}
else
{
hou0[i] = hou0[i + 1];
hou1[i] = hou1[i + 1] + 1;
}
}
ll minn=cnt;
for (ll i = 1; i <= n; i++)
{
if(cnt-pre0[i]>=pre1[i])
{
ll res=cnt-pre0[i]-pre1[i];
ll pos1=n-res+1;
minn=min(minn,cnt-pre0[i]-hou0[pos1]);
}
}
for (ll i = n; i >= 1; i--)
{
if(cnt-hou0[i]>=hou1[i])
{
ll res=cnt-hou0[i]-hou1[i];
ll pos1=res;
minn=min(minn,cnt-pre0[pos1]-hou0[i]);
}
}
printf("%lld\n",minn);
}
int main()
{
scanf("%lld", &T);
while (T--)
{
solve();
}
}
Tips: 语言表达能力有限,欢迎各位聚聚提出建议、问题,有不懂的地方也欢迎提问,我会尽力解答
|