牛客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;
}