BZOJ3747: [POI2015]Kinoman

3747: [POI2015]Kinoman

Time Limit: 30 Sec Memory Limit: 128 MB
Submit: 54 Solved: 28
[Submit][Status]

Description

共有m部电影，编号为1~m，第i部电影的好看值为w[i]。
在n天之中（从1~n编号）每天会放映一部电影，第i天放映的是第f[i]部。
你可以选择l,r(1<=l<=r<=n)，并观看第l,l+1,…,r天内所有的电影。如果同一部电影你观看多于一次，你会感到无聊，于是无法获得这部电影的好看值。所以你希望最大化观看且仅观看过一次的电影的好看值的总和。

Input

第一行两个整数n,m(1<=m<=n<=1000000)。
第二行包含n个整数f[1],f[2],…,f
(1<=f[i]<=m)。
第三行包含m个整数w[1],w[2],…,w[m](1<=w[j]<=1000000)。

Output

输出观看且仅观看过一次的电影的好看值的总和的最大值。

Sample Input

9 4

2 3 1 1 4 1 2 4 1

5 3 6 6

Sample Output

15

样例解释：

观看第2,3,4,5,6,7天内放映的电影，其中看且仅看过一次的电影的编号为2,3,4。

HINT

Source

鸣谢Jcvb

题解：
总算yy出了正解。。。
100W的数据让人比较蛋疼，因为不知道带不带log，不带的话应该是单调队列神马的，发现好像固定右端点，最优左端点并不单调，然后。。。
今天早上跑操的时候忽然想到如果固定右端点，我们能在logn内算出所有l-i（l<=i)的收益，然后取最大的话就可以？
然后考虑加了一个元素a[i]，那么从上一个a[i]出现的位置fa[i]+1到i收益全部增加b[a[i]]，从fa[fa[i]]+1到fa[i]的收益全部减少-b[a[i]]，顺便查询最大值，然后就ok了。
线段树 区间修改+最值查询。
代码：

#include<cstdio>

#include<cstdlib>

#include<cmath>

#include<cstring>

#include<algorithm>

#include<iostream>

#include<vector>

#include<map>

#include<set>

#include<queue>

#include<string>

#define inf 1000000000

#define maxn 1000000+5

#define maxm 500+100

#define eps 1e-10

#define ll long long

#define pa pair<int,int>

#define for0(i,n) for(int i=0;i<=(n);i++)

#define for1(i,n) for(int i=1;i<=(n);i++)

#define for2(i,x,y) for(int i=(x);i<=(y);i++)

#define for3(i,x,y) for(int i=(x);i>=(y);i--)

#define mod 1000000007

using namespace std;

inline int read()

{

int x=0,f=1;char ch=getchar();

while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}

while(ch>='0'&&ch<='9'){x=10*x+ch-'0';ch=getchar();}

return x*f;

}
struct seg{int l,r;ll mx,tag;}t[4*maxn];
int n,m,a[maxn],b[maxn],fa[maxn],last[maxn];
void build(int k,int l,int r)
{
t[k].l=l;t[k].r=r;int mid=(l+r)>>1;t[k].mx=0;t[k].tag=0;
if(l==r)return;
build(k<<1,l,mid);build(k<<1|1,mid+1,r);
}
inline void pushup(int k)
{
t[k].mx=max(t[k<<1].mx,t[k<<1|1].mx);
}
inline void update(int k,ll x)
{
t[k].tag+=x;
t[k].mx+=x;
}
inline void pushdown(int k)
{
if(!t[k].tag)return;
update(k<<1,t[k].tag);
update(k<<1|1,t[k].tag);
t[k].tag=0;
}
ll add(int k,int x,int y,int z)
{
int l=t[k].l,r=t[k].r,mid=(l+r)>>1;
if(l==x&&r==y){update(k,z);return t[k].mx;}
pushdown(k);
ll tmp=0;
if(y<=mid)tmp=add(k<<1,x,y,z);
else if(x>mid)tmp=add(k<<1|1,x,y,z);
else tmp=max(add(k<<1,x,mid,z),add(k<<1|1,mid+1,y,z));
pushup(k);
return tmp;
}

int main()

{

freopen("input.txt","r",stdin);

freopen("output.txt","w",stdout);

n=read();m=read();
for1(i,n)a[i]=read();
for1(i,m)b[i]=read();
for1(i,n)fa[i]=last[a[i]],last[a[i]]=i;
build(1,1,n);
ll ans=0;
for1(i,n)
{
ans=max(ans,add(1,fa[i]+1,i,b[a[i]]));
if(fa[i])ans=max(ans,add(1,fa[fa[i]]+1,fa[i],-b[a[i]]));
}
printf("%lld\n",ans);

return 0;

}

View Code
我线段树写得这么丑居然能排rank5？其他人代码那么短什么情况。。。