牛客周赛 Round 9 （c题题解）

一、题目（小美的01串翻转）

链接：登录—专业IT笔试面试备考平台_牛客网
来源：牛客网
 

小美定义一个 01 串的权值为：每次操作选择一位取反，使得相邻字符都不相等的最小操作次数。
例如，"10001"的权值是 1，因为只需要修改一次：对第三个字符取反即可。
现在小美拿到了一个 01 串，她希望你求出所有非空连续子串的权值之和，你能帮帮她吗？

输入描述:

一个仅包含'0'和'1'的字符串，长度不超过 2000。

输出描述:

所有非空子串的权值和。

示例1

输入

10001

输出

8

说明

长度为 2 的子串中，有 2 个"00"的权值是 1。

长度为 3 的 3 个子串权值都是 1。

长度为 4 的 2 个子串权值都是 1。

长度为 5 的 1 个子串权值是 1。

总权值之和为 2+3+2+1=8

二、思路

每个字符串均有两种情况可以改变，第一种改成以1为开头的01串，第二种改为以0为开头的01串。

a[i]记录开头为1的情况，b[i]记录开头为0 的情况

利用f[j+i-1]-f[j-1] ，代表长度为i的连续的01子串

ans+=(a[i+j-1]-a[j-1],b[i+j-1]-b[j-1])取其最小情况，可以找到这段子串的最小操作次数，也就是该子串的权值

三、代码

#include<bits/stdc++.h>
#define endl '\n'
#define int long long
#define IOS ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
using namespace std;
const int N=2010;
const int inf=0x3f3f3f3f;
char c[N];
int a[N],b[N]; 
void solve()
{
	cin>>c+1;
	int n=strlen(c+1);
	int g=0;
	int i,j,ans=0;
	for(i=1;i<=n;i++)//以1开头 
	{
		g=1-g;
		if(c[i]-'0'!=g) 
		    a[i]=1;
	}
	g=1; 
	for(i=1;i<=n;i++)//以0开头 
	{
		g=1-g; 
		if(c[i]-'0'!=g)
		    b[i]=1; 
	} 
	for(i=1;i<=n;i++)
	{
		a[i]+=a[i-1];
		b[i]+=b[i-1];
	}
	for(i=2;i<=n;i++)
	{
		for(j=1;j+i-1<=n;j++)
		{
			ans+=min(a[j+i-1]-a[j-1],b[j+i-1]-b[j-1]);
		} 
	}
	cout<<ans<<endl;
}
signed main()
{
    int t=1;
    while(t--)
    {
       solve();
    }
    return 0;
}