[Codeforces 489E] Nastya and King-Shamans

[题目链接]

http://codeforces.com/contest/992/problem/E

[算法]

线段树 + 二分

时间复杂度 : O(NlogN^2)

[代码]

#include<bits/stdc++.h>
using namespace std;
const int MAXN = 2e5 + ;
typedef long long ll;

struct Node
{
    int l,r;
    ll mx,sum;
} Tree[MAXN << ];

int i,n,q,x,y,cur,tmp,ans;
ll value[MAXN];
ll pre;

template <typename T> inline void read(T &x)
{
    int f = ; x = ;
    char c = getchar();
    for (; !isdigit(c); c = getchar())
    {
        if (c == '-') f = -f;
    }
    for (; isdigit(c); c = getchar()) x = (x << ) + (x << ) + c - '';
    x *= f;
}
inline void update(int index)
{
    Tree[index].mx = max(Tree[index << ].mx,Tree[index <<  | ].mx);
    Tree[index].sum = Tree[index << ].sum + Tree[index <<  | ].sum;
}
inline void build(int index,int l,int r)
{
    int mid;
    Tree[index].l = l;
    Tree[index].r = r;
    if (l == r)
    {
        Tree[index].mx = value[l];
        Tree[index].sum = value[l];
        return;
    }
    mid = (l + r) >> ;
    build(index << ,l,mid);
    build(index <<  | ,mid + ,r);
    update(index);
}
inline void modify(int index,int pos,int val)
{
    int mid;
    if (Tree[index].l == Tree[index].r)
    {
        Tree[index].mx = Tree[index].sum = val;
        return;
    }
    mid = (Tree[index].l + Tree[index].r) >> ;
    if (mid >= pos) modify(index << ,pos,val);
    else modify(index <<  | ,pos,val);
    update(index);
}
inline int query(int index,int l,int r,ll val)
{
    int mid,tmp;
    if (Tree[index].l == l && Tree[index].r == r)
    {
        if (Tree[index].mx < val) return -;
        if (l == r) return l;
        mid = (Tree[index].l + Tree[index].r) >> ;
        if (Tree[index << ].mx >= val) return query(index << ,l,mid,val);
        else return query(index <<  | ,mid + ,r,val);
    }
    mid = (Tree[index].l + Tree[index].r) >> ;
    if (mid >= r) tmp = query(index << ,l,r,val);
    else if (mid +  <= l) tmp = query(index <<  | ,l,r,val);
    else
    {
        tmp = query(index << ,l,mid,val);
        if (tmp != -) return tmp;
        return query(index <<  | ,mid + ,r,val);
    }
    return tmp;
}
inline ll query_sum(int index,int l,int r)
{
    int mid;
    if (Tree[index].l == l && Tree[index].r == r) return Tree[index].sum;
    mid = (Tree[index].l + Tree[index].r) >> ;
    if (mid >= r) return query_sum(index << ,l,r);
    else if (mid +  <= l) return query_sum(index <<  | ,l,r);
    else return query_sum(index << ,l,mid) + query_sum(index <<  | ,mid + ,r);
}
int main()
{

    read(n); read(q);
    for (i = ; i <= n; i++) read(value[i]);
    build(,,n);
    while (q--)
    {
        read(x); read(y);
        value[x] = y;
        modify(,x,y);
        if (value[] == )
        {
            printf("1\n");
            continue;
        }
        cur = pre = tmp = ; ans = -;
        while (cur < n)
        {
            tmp = query(,cur + ,n,pre);
            if (tmp == -) break;
            cur = tmp;
            pre = query_sum(,,tmp);
            if (pre - value[cur] == value[cur])
            {
                ans = tmp;
                break;
            }
        }
        printf("%d\n",ans);
    }

    return ;
}