3747: [POI2015]Kinoman 线段树

最新推荐文章于 2018-12-13 11:29:08 发布

ws_fqk

最新推荐文章于 2018-12-13 11:29:08 发布

阅读量479

点赞数

分类专栏： My Code

本文链接：https://blog.csdn.net/Phenix_2015/article/details/50486776

版权

My Code 专栏收录该内容

349 篇文章 0 订阅

订阅专栏

QAQ看到N=100w以为有O(n)作法，思考无果，准备查题解的时候，一眼瞄到了时间限制60sec。。于是想了想O(nlogn)的作法，感觉可以试试。。于是写了一发线段树26s过了QAQ。

作法比较简单，首先记录每个点上一次出现的位置。然后加入这个点的时候(pre[i],i]都+w[f[i]],(pre[pre[i]],pre[i]]都-w[f[i]],每次询问(pre[i],i]的区间最值更新答案。。显然线段树。

#include<bits/stdc++.h>
#define ll long long 
#define N 1000005
using namespace std;
int n,m;
ll ans;
int pre[N],pos[N],w[N],f[N];
int l[N<<2],r[N<<2];
ll tag[N<<2],mx[N<<2];
inline int read()
{
    int a=0,f=1; char c=getchar();
    while (c<'0'||c>'9') {if (c=='-') f=-1; c=getchar();}
    while (c>='0'&&c<='9') {a=a*10+c-'0'; c=getchar();}
    return a*f;
}
inline void pushup(int k)
{
    mx[k]=max(mx[k<<1],mx[k<<1|1]);
}
inline void pushdown(int k)
{
    if (!tag[k]||l[k]==r[k]) return;
    tag[k<<1]+=tag[k]; tag[k<<1|1]+=tag[k];
    mx[k<<1]+=tag[k]; mx[k<<1|1]+=tag[k];
    tag[k]=0;
}
void build(int k,int x,int y)
{
    l[k]=x; r[k]=y;
    if (l[k]==r[k]) return;
    int mid=l[k]+r[k]>>1;
    build(k<<1,x,mid); build(k<<1|1,mid+1,y);
}
void add(int k,int x,int y,int val)
{
    if (x>y) return;
    if (l[k]==x&&r[k]==y)
    {
        tag[k]+=val;
        mx[k]+=val;
        return;
    }
    pushdown(k);
    int mid=l[k]+r[k]>>1;
    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);
    pushup(k);
}
ll query(int k,int x,int y)
{
    if (x>y) return 0;
    if (l[k]==x&&r[k]==y) return mx[k];
    pushdown(k);
    int mid=l[k]+r[k]>>1;
    if (y<=mid) return query(k<<1,x,y);
    else if (x>mid) return query(k<<1|1,x,y);
    else return max(query(k<<1,x,mid),query(k<<1|1,mid+1,y));
}
int main()
{
    n=read(); m=read();
    for (int i=1;i<=n;i++) 
        f[i]=read(),pre[i]=pos[f[i]],pos[f[i]]=i;
    for (int i=1;i<=m;i++) w[i]=read();
    build(1,1,n);
    for (int i=1;i<=n;i++)
    {
        add(1,pre[i]+1,i,w[f[i]]);
        add(1,pre[pre[i]]+1,pre[i],-w[f[i]]);
        ans=max(ans,query(1,pre[i]+1,i));
    }
    cout << ans;
    return 0;
}