【前缀和】 洛谷1371 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,对于每一个O,假设在x位置,它所产生的NOI数量=N[x]*(I[n]-I[x])

则共可生成sum个NOI串

当添加N的时候，一定在最前面添加新增的NOI会最多,为ans1

当添加I的时候，一定在最后面添加新增的NOI 会最多，为ans3

当添加O的时候，要枚举O 在每一个位置的情况，从每个位置新增加的最大值中选出最大的情况为ans2

比较ans1,2,3的大小  设最大值为res

输出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;

}