洛谷 P1371 NOI元丹

题目描述
小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，用前缀和可知n和i的个数。

---

#include<algorithm>#include<iostream>#include<cstdio>using namespace std;const int N=100005;int n,s[N][3];long long ans,res,tmp;char ch[N];int main(){    scanf("%d%s",&n,ch+1);    for(int i=1;i<=n;i++)    {        s[i][0]=s[i-1][0],s[i][1]=s[i-1][1],s[i][2]=s[i-1][2];        if(ch[i]=='N')            s[i][0]++;        else if(ch[i]=='O')            s[i][1]++;        else            s[i][2]++;    }    for(int i=1;i<=n;i++)        if(ch[i]=='O')            ans+=(long long)s[i][0]*(s[n][2]-s[i][2]);    tmp=0;    for(int i=1;i<=n;i++)//n        if(ch[i]=='O')            tmp+=(long long)(s[n][2]-s[i][2]);    if(tmp>res)        res=tmp;    tmp=0;    for(int i=1;i<=n;i++)//i        if(ch[i]=='O')            tmp+=(long long)s[i][0];    if(tmp>res)        res=tmp;    tmp=0;    for(int i=1;i<=n;i++)//o    {        tmp=(long long)s[i][0]*(s[n][2]-s[i][2]);        if(tmp>res)            res=tmp;    }    printf("%lld\n",ans+res);    return 0;}