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[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

题解：
总算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了。
线段树 区间修改+最值查询。
代码：
  1 #include<cstdio>
  2 
  3 #include<cstdlib>
  4 
  5 #include<cmath>
  6 
  7 #include<cstring>
  8 
  9 #include<algorithm>
 10 
 11 #include<iostream>
 12 
 13 #include<vector>
 14 
 15 #include<map>
 16 
 17 #include<set>
 18 
 19 #include<queue>
 20 
 21 #include<string>
 22 
 23 #define inf 1000000000
 24 
 25 #define maxn 1000000+5
 26 
 27 #define maxm 500+100
 28 
 29 #define eps 1e-10
 30 
 31 #define ll long long
 32 
 33 #define pa pair<int,int>
 34 
 35 #define for0(i,n) for(int i=0;i<=(n);i++)
 36 
 37 #define for1(i,n) for(int i=1;i<=(n);i++)
 38 
 39 #define for2(i,x,y) for(int i=(x);i<=(y);i++)
 40 
 41 #define for3(i,x,y) for(int i=(x);i>=(y);i--)
 42 
 43 #define mod 1000000007
 44 
 45 using namespace std;
 46 
 47 inline int read()
 48 
 49 {
 50 
 51     int x=0,f=1;char ch=getchar();
 52 
 53     while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
 54 
 55     while(ch>='0'&&ch<='9'){x=10*x+ch-'0';ch=getchar();}
 56 
 57     return x*f;
 58 
 59 }
 60 struct seg{int l,r;ll mx,tag;}t[4*maxn];
 61 int n,m,a[maxn],b[maxn],fa[maxn],last[maxn];
 62 void build(int k,int l,int r)
 63 {
 64     t[k].l=l;t[k].r=r;int mid=(l+r)>>1;t[k].mx=0;t[k].tag=0;
 65     if(l==r)return;
 66     build(k<<1,l,mid);build(k<<1|1,mid+1,r);
 67 }
 68 inline void pushup(int k)
 69 {
 70     t[k].mx=max(t[k<<1].mx,t[k<<1|1].mx);
 71 }
 72 inline void update(int k,ll x)
 73 {
 74     t[k].tag+=x;
 75     t[k].mx+=x;
 76 }
 77 inline void pushdown(int k)
 78 {
 79     if(!t[k].tag)return;
 80     update(k<<1,t[k].tag);
 81     update(k<<1|1,t[k].tag);
 82     t[k].tag=0;
 83 }
 84 ll add(int k,int x,int y,int z)
 85 {
 86     int l=t[k].l,r=t[k].r,mid=(l+r)>>1;
 87     if(l==x&&r==y){update(k,z);return t[k].mx;}
 88     pushdown(k);
 89     ll tmp=0;
 90     if(y<=mid)tmp=add(k<<1,x,y,z);
 91     else if(x>mid)tmp=add(k<<1|1,x,y,z);
 92     else tmp=max(add(k<<1,x,mid,z),add(k<<1|1,mid+1,y,z));
 93     pushup(k);
 94     return tmp;
 95 }
 96 
 97 int main()
 98 
 99 {
100 
101     freopen("input.txt","r",stdin);
102 
103     freopen("output.txt","w",stdout);
104 
105     n=read();m=read();
106     for1(i,n)a[i]=read();
107     for1(i,m)b[i]=read();
108     for1(i,n)fa[i]=last[a[i]],last[a[i]]=i;
109     build(1,1,n);
110     ll ans=0;
111     for1(i,n)
112     {
113         ans=max(ans,add(1,fa[i]+1,i,b[a[i]]));
114         if(fa[i])ans=max(ans,add(1,fa[fa[i]]+1,fa[i],-b[a[i]]));
115     }
116     printf("%lld\n",ans);
117 
118     return 0;
119 
120 }
View Code

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

posted @ 2014-11-18 11:01  ZYF-ZYF  Views(731)  Comments(3)  Edit  收藏  举报