做法
对于寻找一个"aab"的子序列只要枚举“b”,并在'b'之前的'a'中取两个就能求出子序列的数量。
因为可以在任意位置中插入一个字符。很容易想到最优的插法:1.在最前面插入'a';2.在最后面插入'b'
在这两种方案中取到最大的一种即可
注意:答案会爆long long
代码
// Problem: 子序列 // Contest: NowCoder // URL: https://ac.nowcoder.com/acm/contest/11233/B // Memory Limit: 1048576 MB // Time Limit: 2000 ms // // Powered by CP Editor (https://cpeditor.org) #include <bits/stdc++.h> using namespace std; #define pb push_back #define mp(aa,bb) make_pair(aa,bb) #define _for(i,b) for(int i=(0);i<(b);i++) #define rep(i,a,b) for(int i=(a);i<=(b);i++) #define per(i,b,a) for(int i=(b);i>=(a);i--) #define mst(abc,bca) memset(abc,bca,sizeof abc) #define X first #define Y second #define lowbit(a) (a&(-a)) #define debug(a) cout<<#a<<":"<<a<<"\n" typedef long long ll; typedef pair<int,int> pii; typedef unsigned long long ull; typedef long double ld; const int N=100010; const int INF=0x3f3f3f3f; const int mod=1e9+7; const double eps=1e-6; const double PI=acos(-1.0); __int128 cnt,ans1,ans2; __int128 C(ll n,ll m){ ll ans = 1; for(ll i=1;i<=m;i++){ ans=ans*(n-m+i)/i; //注意一定要先乘再除 } return ans; } void write(__int128 x) { if (x < 0) { x = -x; putchar('-'); } if (x > 9) write(x / 10); putchar(x % 10 + '0'); } void solve(){ string s;cin>>s; string ss=s+'b'; for(char x:ss){ if(x=='a') cnt++; else ans1+=C(cnt,2); } ss='a'+s; cnt=0; for(char x:ss){ if(x=='a') cnt++; else ans2+=C(cnt,2); } write(max(ans1,ans2)); } int main(){ // int t;cin>>t;while(t--) solve(); return 0; }