bzoj3747-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],…,fn。
第三行包含m个整数w[1],w[2],…,wm。

---

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。

---

题解

---

思路分析：这道题运用了很巧妙的方法：移端点！
考虑移动左端点，计算最优右端点。
对于某一种电影i，这种电影的第一次出现的位置记为last[i]。
nxt[h]记录第h天的电影下次播放时间。
考虑左端点在i。考虑该天的电影k=f[i]，那么右端点在[i,nxt[i]-1]的时候会得到这部电影的好看值。这是一个区间加操作。
接下来枚举左端点，移动到i+1。
！！那么注意到，右端点在[i,nxt[i]-1]无法获得k电影的好看值了，区间减法。而在[nxt[i],nxt[nxt[i]]-1]处又可以获得这个好看值了，区间加法。

---

代码

#include<cstdio>
#include<cmath>
#include<cstring>
#include<iostream>
#define ll long long
using namespace std;
int read()
{
    int x=0;char ch=getchar();
    while(ch>'9'||ch<'0')ch=getchar();
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x;
}
int n,m;
int f[1000005],w[1000005];
int last[1000005],nxt[1000005];
ll ans;
struct seg{
    int l,r;
    ll tag,mx;
}t[4000005];
void pushdown(int k)
{
    int l=t[k].l,r=t[k].r;
    if(l==r)return;
    ll tag=t[k].tag;t[k].tag=0;
    t[k<<1].tag+=tag;t[k<<1|1].tag+=tag;
    t[k<<1].mx+=tag;t[k<<1|1].mx+=tag;
}
void build(int k,int l,int r)
{
    t[k].l=l;t[k].r=r;
    if(l==r)return;
    int mid=(l+r)>>1;
    build(k<<1,l,mid);build(k<<1|1,mid+1,r);
}
void add(int k,int x,int y,int val)
{
    if(t[k].tag)pushdown(k);
    int l=t[k].l,r=t[k].r,mid=(l+r)>>1;
    if(l==x&&y==r)
    {
        t[k].tag=val;t[k].mx+=val;
        return;
    }
    if(y<=mid)add(k<<1,x,y,val);
    else if(x>mid)add(k<<1|1,x,y,val);
    else add(k<<1,x,mid,val),add(k<<1|1,mid+1,y,val);
    t[k].mx=max(t[k<<1].mx,t[k<<1|1].mx);
}
int main()
 {
     n=read();m=read();
     for(int i=1;i<=n;i++)f[i]=read();
     for(int i=1;i<=m;i++)w[i]=read();
     for(int i=n;i;i--)
     {
         nxt[i]=last[f[i]];
         last[f[i]]=i;
     }
     build(1,1,n);
     for(int i=1;i<=m;i++)
         if(last[i])
         {
             if(!nxt[last[i]])add(1,last[i],n,w[i]);
             else add(1,last[i],nxt[last[i]]-1,w[i]);
         }
     for(int i=1;i<=n;i++)
     {
         ans=max(ans,t[1].mx);
         int t=nxt[i];
         if(t)
         {
             add(1,i,t-1,-w[f[i]]);
             if(nxt[t])add(1,t,nxt[t]-1,w[f[i]]);
             else add(1,t,n,w[f[i]]);
         }
         else add(1,i,n,-w[f[i]]);
     }
     printf("%lld\n",ans);
     return 0;
}