Bzoj3747 [POI1015] Kinoman

 

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[n](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。

 

Bzoj权限题。只过了样例，尚未评测。

题面搬运自神犇hzwer博客。

 

 

线段树维护区间最值，枚举区间起点，不断把起点前面的电影贡献值删去，再把当前电影“从当前直到它下次出现之前”的区间都新加上当前电影的贡献值，不断更新最大答案。

刚开始按区间加法写的，所以变量用了sum，写到一半发现只搞区间最大值就好，然而懒得改变量名了……

 

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#define ls l,mid,rt<<1
#define rs mid+1,r,rt<<1|1
#define LL long long
using namespace std;
const int mxn=1000010;
int n,m;
int f[mxn],w[mxn];
int next[mxn],last[mxn];
int cnt[mxn];
struct node{
    LL sum;//变量名是sum，实际是存区间最大值 
    LL tag;
}t[mxn<<2];
void Build(int l,int r,int rt){//init //实际并没有用到 
    if(l==r){
        t[rt].sum=t[rt].tag=0;
        return;
    }
    int mid=(l+r)>>1;
    Build(ls);Build(rs);
    return;
}
void pushdown(int l,int r,int rt){
    if(l==r)return;
    if(t[rt].tag){ 
        int mid=(l+r)>>1;
        t[rt<<1].tag+=t[rt].tag;
        t[rt<<1|1].tag+=t[rt].tag;
        t[rt<<1].sum+=t[rt].tag;
        t[rt<<1|1].sum+=t[rt].tag;
        t[rt].tag=0;
    }
}
void add(int L,int R,int v,int l,int r,int rt){
    if(t[rt].tag)pushdown(l,r,rt);
    if(L<=l && r<=R){
        t[rt].tag+=v;
        t[rt].sum+=v;
        return;
    }
    int mid=(l+r)>>1;
    if(L<=mid)add(L,R,v,ls);
    if(R>mid)add(L,R,v,rs);
    t[rt].sum=max(t[rt<<1].sum,t[rt<<1|1].sum);
    return;
}
void read(){
    scanf("%d%d",&n,&m);
    int i,j;
    for(i=1;i<=n;i++)scanf("%d",&f[i]);
    for(i=1;i<=m;i++)scanf("%d",&w[i]);
    return;
}
int main(){
    read();
    int i,j;
    for(i=n;i;i--){
        next[i]=last[f[i]];
        last[f[i]]=i;
    }
    for(i=1;i<=m;i++){
        if(last[i]){
            if(!next[last[i]])add(last[i],n,w[i],1,n,1); 
            else add(last[i],next[last[i]]-1,w[i],1,n,1);
        }
    }
    LL ans=0;
    for(i=1;i<=n;i++){
        ans=max(ans,t[1].sum);
        int tmp=next[i];
        if(tmp){
            add(i,tmp-1,-w[f[i]],1,n,1);
            if(next[tmp])add(tmp,next[tmp]-1,w[f[i]],1,n,1); 
            else add(tmp,n,w[f[i]],1,n,1);//如果之后没有相同电影了，从next[i]到n的区间都加值 
        }
        else add(i,n,-w[f[i]],1,n,1);//删除上一个区间的值 
    }
    printf("%lld",ans);
    return 0;
}