【BZOJ 3747】 3747: [POI2015]Kinoman （线段树）

3747: [POI2015]Kinoman

Time Limit: 60 Sec  Memory Limit: 128 MB
Submit: 830  Solved: 338

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。

HINT

Source

鸣谢Jcvb

【分析】

　　这题应该挺经典。？

　　就是先弄一个next，然后每次求以i结尾的最大值。

　　i的值为w，next[i]的值为-w，更前面的next的值为0，线段树维护这个，（好像树状数组也是可以的），然后就好了。

 #include<cstdio>
 #include<cstdlib>
 #include<cstring>
 #include<iostream>
 #include<algorithm>
 using namespace std;
 #define Maxn 1000010
 #define LL long long

 // int mymax(int x,int y) {return x>y?x:y;}
 LL mymax(LL x,LL y) {return x>y?x:y;}

 int a[Maxn],w[Maxn],f[Maxn],ft[Maxn],nt[Maxn];

 struct node
 {
     int l,r,lc,rc;
     LL mx,lazy;
 }tr[Maxn*];

 int tot=;
 int build(int l,int r)
 {
     int x=++tot;
     tr[x].l=l;tr[x].r=r;
     tr[x].lazy=tr[x].mx=;
     if(l<r)
     {
         int mid=(l+r)>>;
         tr[x].lc=build(l,mid);
         tr[x].rc=build(mid+,r);
     }
     else tr[x].lc=tr[x].rc=;
     return x;
 }

 void upd(int x)
 {
     tr[x].mx+=tr[x].lazy;
     if(tr[x].lazy==||tr[x].l==tr[x].r) {tr[x].lazy=;return;}
     int lc=tr[x].lc,rc=tr[x].rc;
     tr[lc].lazy+=tr[x].lazy;
     tr[rc].lazy+=tr[x].lazy;
     tr[x].lazy=;
 }

 void change(int x,int l,int r,int y)
 {
     if(tr[x].l==l&&tr[x].r==r)
     {
         tr[x].lazy+=y;
         upd(x);
         return;
     }
     upd(x);
     int mid=(tr[x].l+tr[x].r)>>;
     if(r<=mid) change(tr[x].lc,l,r,y);
     else if(l>mid) change(tr[x].rc,l,r,y);
     else
     {
         change(tr[x].lc,l,mid,y);
         change(tr[x].rc,mid+,r,y);
     }
     upd(tr[x].lc);upd(tr[x].rc);
     tr[x].mx=mymax(tr[tr[x].lc].mx,tr[tr[x].rc].mx);
 }

 int main()
 {
     int n,m;
     scanf("%d%d",&n,&m);
     for(int i=;i<=n;i++) scanf("%d",&a[i]);
     for(int i=;i<=m;i++) scanf("%d",&w[i]);
     memset(ft,,sizeof(ft));
     for(int i=;i<=n;i++)
     {
         nt[i]=ft[a[i]];
         ft[a[i]]=i;
     }
     build(,n);
     LL maxx=;
     for(int i=;i<=n;i++)
     {
         change(,,i,w[a[i]]);
         if(nt[i]) change(,,nt[i],-*w[a[i]]);
         if(nt[nt[i]]) change(,,nt[nt[i]],w[a[i]]);
         maxx=mymax(maxx,tr[].mx);
     }
     printf("%lld\n",maxx);
     return ;
 }

2017-04-08 10:54:12