codeforces 830B 树状数组

最新推荐文章于 2019-12-27 17:53:05 发布

原创最新推荐文章于 2019-12-27 17:53:05 发布 · 357 阅读

0 ·

CC 4.0 BY-SA版权

树状数组专栏收录该内容

2 篇文章

订阅专栏

本文介绍了一种使用线段树和树状数组实现的区间更新查询算法。该算法能够高效地处理一系列数值的区间加法操作及查询指定位置的累计加法结果。通过对输入数据进行预处理，并利用树状数组来加速区间更新过程，可以显著提高算法效率。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
int read(){int ret=0;char ch=getchar();while(ch<'0'||ch>'9') ch=getchar();for(;ch>='0'&&ch<='9';ch=getchar()) ret=ret*10+ch-'0';return ret;}
vector<int> q[100005];
int sum[100005];
#define lowbit(x) x&(-x)
void add(int x,int p){while(x<=100000){sum[x]+=p;x+=lowbit(x);}}
int get(int x){int ans=0;while(x){ans+=sum[x];x-=lowbit(x);}return ans;}
int main()
{
    int n=read();
    for(int i=1;i<=n;i++)
        q[read()].push_back(i),add(i,1);
    int cur=1;
    ll ans=1;
    for(int i=1;i<=100000;i++)
    {
        int pos=upper_bound(q[i].begin(),q[i].end(),cur-1)-q[i].begin();
        for(int j=pos;j<q[i].size();j++)
        {
           ans+=get(q[i][j])-get(cur);
           cur=q[i][j];
           add(cur,-1);
        }
        for(int j=0;j<pos;j++)
        {
            if(q[i][j]<cur)ans+=get(q[i][j])+(get(100000)-get(cur));
            else ans+=get(q[i][j])-get(cur);
            cur=q[i][j];
            add(cur,-1);
        }
    }
    cout<<ans<<endl;
}