codeforces 830B 树状数组

来源:互联网 发布:淘宝如何设置快递模板 编辑:程序博客网 时间:2024/06/07 07:33
#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;}
阅读全文
2 0
原创粉丝点击