题目描述
小A打算开始炼NOI元丹(什么鬼),据说吃了可以提高NOI时的成绩。
是这么练的。元丹有三种元核,'N','O','I'。现有很多个这样原核,按顺序排成一行。炼元丹时,从左往右分别挑出'N','O','I'三个原核吞下。
现在他关心,有几种服用方式……且慢!
他觉得服用方式太少,以至于不能成仙。所以他可以通过某个途径,得到'N','O','I'的三种原核中的任意一个,至于哪一种由他决定。然后他将获得这个原核的插入到这一排原核中的任意位置(包括最前最后)。
现在你要知道,新的元核序列中能有多少种'N','O','I'的取出方式。子串的字母并不要求连续。
输入输出格式
输入格式:第一行,一个整数N,表示字符串的长度。
第二行,一行字符串,里面只有只有'N','O','I'三种字母。
输出格式:表示出最多可以提炼出来的NOI元丹的方案种数。
输入输出样例
输入样例#1:
5 NOIOI
输出样例#1:
6
说明
样例解释
他可以获取一个N元核,加到最前面。
NNOIOI | NNOIOI | NNOIOI | NNOIOI | NNOIOI | NNOIOI
~ ~~ | ~ ~ ~ | ~ ~~ | ~~~ | ~~ ~ | ~ ~~
30%的数据N<=200
50%的数据N<=2000
100%的数据3<=N<=100000
分析
对于原串,枚举每一个O,对于每一个O,假设在x位置,它所产生的NOI数量=N[x]*(I[n]-I[x])
则共可生成sum个NOI串
当添加N的时候,一定在最前面添加新增的NOI会最多,为ans1
当添加I的时候,一定在最后面添加新增的NOI 会最多,为ans3
当添加O的时候,要枚举O 在每一个位置的情况,从每个位置新增加的最大值中选出最大的情况为ans2
输出sum+res
附上代码
#include<iostream>
#include<cstdio>
#include<string>
#define MAXN 100000
using namespace std;
long long sum=0;
int n;
long long N[MAXN],O[MAXN],I[MAXN];
long long ans[5];
string origin;
int main()
{
cin>>n>>origin;
if(origin[0]=='N') N[0]=1;
if(origin[0]=='O') O[0]=1;
if(origin[0]=='I') I[0]=1;
for(int i=1; i<n; i++)
{
if(origin[i]=='N')
N[i]=N[i-1]+1;
else
N[i]=N[i-1];
if(origin[i]=='O')
O[i]=O[i-1]+1;
else
O[i]=O[i-1];
if(origin[i]=='I')
I[i]=I[i-1]+1;
else
I[i]=I[i-1];
}
for(int i=0;i<n;i++)
if(origin[i]=='O')
sum+=N[i]*(I[n-1]-I[i]);
for(int i=0;i<n;i++)
if(origin[i]=='O')
ans[1]+=(I[n-1]-I[i]);
for(int i=0;i<n;i++)
if(origin[i]=='O')
ans[3]+=N[i];
for(int i=0;i<n;i++)
ans[2]=max(ans[2],N[i]*(I[n-1]-I[i]));
long long res=0;
for(int i=1;i<=3;i++)
res=max(res,ans[i]);
cout<<res+sum<<endl;
return 0;
}