码迷,mamicode.com
首页 > 其他好文 > 详细

bzoj3747

时间:2018-01-11 19:10:12      阅读:187      评论:0      收藏:0      [点我收藏+]

标签:div   update   技术分享   color   bsp   ace   pac   main   blog   

线段树

线段树记录到每个位置截止的答案

技术分享图片
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 1e6 + 5;
int n, m;
ll ans;
int nxt[N], last[N], a[N], b[N], vis[N];
ll mx[N << 2], tag[N << 2];
void pushdown(int x)
{
    if(!tag[x]) return;
    tag[x << 1] += tag[x];
    tag[x << 1 | 1] += tag[x];
    mx[x << 1] += tag[x];
    mx[x << 1 | 1] += tag[x];
    tag[x] = 0;
}
void update(int l, int r, int x, int a, int b, int d)
{
    if(l > b || r < a) return;
    if(l >= a && r <= b) 
    {
        mx[x] += d;
        tag[x] += d;
        return;
    }
    pushdown(x);
    int mid = (l + r) >> 1;
    update(l, mid, x << 1, a, b, d);
    update(mid + 1, r, x << 1 | 1, a, b, d);
    mx[x] = max(mx[x << 1], mx[x << 1 | 1]);
}
int main()
{
    scanf("%d%d", &n, &m);
    for(int i = 1; i <= n; ++i) 
    {
        scanf("%d", &a[i]);
        if(last[a[i]]) nxt[last[a[i]]] = i; 
        last[a[i]] = i;
    }
    for(int i = 1; i <= m; ++i) scanf("%d", &b[i]);
    for(int i = 1; i <= n; ++i) if(!vis[a[i]]) 
    {
        update(1, n, 1, i, nxt[i] ? nxt[i] - 1 : n, b[a[i]]);
        vis[a[i]] = 1;
    }
    for(int i = 1; i <= n; ++i) 
    {
        ans = max(ans, mx[1]);
        update(1, n, 1, i, nxt[i] ? nxt[i] - 1 : n, -b[a[i]]);
        if(nxt[i]) update(1, n, 1, nxt[i], nxt[nxt[i]] ? nxt[nxt[i]] - 1 : n, b[a[i]]); 
    }
    printf("%lld\n", ans);
    return 0;
}
View Code

 

bzoj3747

标签:div   update   技术分享   color   bsp   ace   pac   main   blog   

原文地址:https://www.cnblogs.com/19992147orz/p/8269909.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!