【bzoj3747】【POI2015】【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。
题解:
对于每种电影记录一下它第一次出现的位置。
对于每一天记录一下下一个和它相同的点的位置。
然后我们扫描左端点即可。
代码:
#include<iostream>
#include<cstdio>
#define N 1000100
using namespace std;
int point[N],next[N],n,m,a[N];
long long ans,t[N*4],f[N],p[N*4];
void paint(int k,int l,int r,long long v){t[k]+=v;p[k]+=v; } 
void pushdown(int k,int l,int r){
  int mid=(l+r)>>1;
  paint(k<<1,l,mid,p[k]);
  paint(k<<1|1,mid+1,r,p[k]); 
  p[k]=0;
}
void add(int k,int l,int r,int ll,int rr,long long v){
   int mid=(l+r)>>1;
   if (ll<=l&&r<=rr){paint(k,l,r,v);return;}
   if (p[k]) pushdown(k,l,r);
   if (ll<=mid) add(k<<1,l,mid,ll,rr,v);
   if (rr>mid) add(k<<1|1,mid+1,r,ll,rr,v);
   t[k]=max(t[k<<1],t[k<<1|1]);  
}
int main(){
  scanf("%d%d",&n,&m);
  for (int i=1;i<=n;i++) scanf("%d",&a[i]);
  for (int i=1;i<=m;i++) scanf("%lld",&f[i]);
  for (int i=n;i>=1;i--){
    next[i]=point[a[i]];
    point[a[i]]=i;
  }
  for (int i=1;i<=m;i++)
    if (point[i]){
       if(next[point[i]])add(1,1,n,point[i],next[point[i]]-1,f[i]);
       else add(1,1,n,point[i],n,f[i]);
    }
  for (int i=1;i<=n;i++){
    ans=max(ans,t[1]);int t=next[i];
    if (t){
      add(1,1,n,i,t-1,-f[a[i]]);
      if (next[t]) add(1,1,n,t,next[t]-1,f[a[i]]);
      else add(1,1,n,t,n,f[a[i]]);
    }
    else add(1,1,n,i,n,-f[a[i]]); 
  }
  cout<<ans<<endl;  
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值