B 子序列

 牛客IOI周赛26-普及组：B 子序列

 

链接：https://ac.nowcoder.com/acm/contest/11233/B

 

给出一个仅包含 a,b 的字符串 A。在 A 中间任意位置（包括开头结尾）插入一个字符，最大化 aab 作为子序列（可以不连续）在 A 中出现的次数。

输入描述:

 

第一行一个仅包含 a,b 的字符串 A。

 

输出描述:

 输出一个整数，为插入一个字符后，aab 作为子序列在 A 中出现的次数的最大值。 

输入

abababa

输出

10

说明

在第一个字符后插入一个 a，变为 aabababa。

示例2

输入

ababbaababa

输出

33

输入

aa

输出

1

备注:

设 n 为 A 的长度。
对于20% 的数据，n≤10。 

对于 40% 的数据，n≤1000。

对于另外 10% 的数据，A 中只有 a。

对于另外 10% 的数据，A 中只有 b。

对于 90% 的数据， n≤2×。
对于 100% 的数据， 1≤n≤5.5×

 

知道最优方案后，主要的问题就是计数了，计数的关键在于找到 b 字符前有多少个 a 字符，然后算Cn2；

用样例2演示一下吧；

按照以上方法试了一遍，发现有一个大数据样例不过，试了好久还是没搞出来，结束之后才发现要用到 unsigned long long ; 

#include<bits/stdc++.h>
using namespace std;
#define ll unsigned long long

ll c(ll a)
{
	return a*(a-1)/2;
}

int main()
{
	string a;
	cin>>a;
	string b = a + 'b';
	ll sum = 1;
	ll cnt1 = 0;
	for(int i=0; i<a.length(); i++)
	{
		if(a[i]=='a')
			sum++;
		if(a[i]=='b')
			cnt1+=c(sum);
	}
	sum=0;
	ll cnt2=0;
	for(int i=0; i<b.length(); i++)
	{
		if(b[i]=='a')
			sum++;
		if(b[i]=='b')
			cnt2+=c(sum);
	}
	cnt1>cnt2?cout<<cnt1:cout<<cnt2;
}