CODEFORCES, 265E Choosing Balls

题意：有n个球，每个球有一个值和一种颜色，给出每个球的值和颜色的序列，不改变每个球的相对顺序，从中选出一些球使得sum最大。

sum的计算 如果每个球前面的球颜色和自己相同，则sum+=a* val，否则sum+=b*val 。val是球自身的值。如果球在第一个即前面无球，则sum+=b*val。

然后给出q次查询，每次给定a，b的值然后求sum最大。(可以一个也不选 sum为0)

1 ≤ n ≤ 10⁵; 1 ≤ q ≤ 500

策略：如果我们已经知道了前面i个球，以颜色c结尾的最大sum，那么我们在取第i+1个颜色为col球的时候，我们要做的就是

1.找到前i个球中以颜色为col结尾的sum最大 用sum+a*val去更新 前i+1个球以颜色col结尾的最大sum

2.找到前i个球中以其他颜色结尾的球sum最大 用sum+b*val去更新 前i+1个球以颜色col结尾的最大sum

3.如若该求为所选球的第一个球 则用b*val更新。

状态记为 dp[j][j] 表示前i个球以颜色j结尾的球最大的sum。

对于找到颜色不为col结尾的最大sum，我们可能会想到用线段树去维护每个颜色结尾的最大sum。

这样我们的时间复杂度在q*n* logn中可以完成规定操作。但是这个似乎还不够好，时间卡的非常紧。

于是在放弃线段树的情况下我们思考如何快速找到 我们所需要的两个最大sum。既然是2个最大我们会很自然的想到用最大值和次大值来记录这些sum即可，其他的值对于我们来说是没有用的。所以在每次更新时，如果最大值对应自身的颜色，那么我们用次大值去+b*val更新 否则就用最大值。最大与次大O（1）即可完成维护。

所以复杂度为 q*n 。

//Author: Wenjun Shi
//on 2016/7/14
#include <iostream>
#include <cstring>
#include <utility>
#include <cstdio>
#define MAX_N 100005
using namespace std;
typedef long long ll;
pair<ll,ll> Max,Cmax;
ll color[MAX_N];
ll val[MAX_N];
ll bc[MAX_N];
ll a,b;

void update(ll v,ll i)
{
    if(v>Max.first&&bc[i]==Max.second)
    {
        Max.first=v;
    }
    else if(v>Max.first&&bc[i]!=Max.second)
    {
        Cmax=Max;
        Max.first=v;
        Max.second=bc[i];
    }
    else if(v>Cmax.first&&bc[i]!=Max.second)
    {
        Cmax.first=v;
        Cmax.second=bc[i];
    }
}

ll take(ll x) {
	ll ans = color[bc[x]] + (ll) a * val[x];
	if(Max.second != bc[x]) return max(ans, Max.first + b * (ll) val[x]);
	return max(ans, Cmax.first+ b * (ll) val[x]);
}

int main()
{
    ll n,q;
    cin>>n>>q;
    for(int i=1;i<=n;i++) scanf("%I64d",&val[i]);
    for(int i=1;i<=n;i++) scanf("%I64d",&bc[i]);
    for(int j=0;j<q;j++)
    {
        scanf("%I64d %I64d",&a,&b);
        memset(color,0x80,sizeof(color));
        Max.first=Cmax.first=0;
        Max.second=Cmax.second=-1;
        for(int i=1;i<=n;i++)
        {
            ll ans = (ll) b * val[i];
           // ll ans=0;
			ans = max(ans, take(i));
			color[bc[i]] = max(color[bc[i]], ans);
			update(color[bc[i]], i);
        }
        cout<<Max.first<<endl;
    }
    return 0;
}