Codeforces 831 E. Cards Sorting set+树状数组模拟

传送门：E. Cards Sorting

题意：给你n张卡片，每张卡片上都有数字，每次从顶端抽一张，如果是当前剩余卡片中最小的，就把他扔掉，否则放到堆底，问抽多少次能把所有卡片都扔掉。

正解：将卡片上的数字的对应初始位置用set记录下来，大小相同数字的位置放到一个set里，并且用树状数组维护每个位置的卡片是否还在，然后我们从最小的数字开始遍历，假设上一次停下的位置是last，当前最小数字是x，那么我们就在set[x]里找大于last的第一个位置pos，set[x]中如果还有小于pos的数，说明我们要从last遍历到n（当然是只遍历剩余的卡片，这时候树状数组的作用就体现出来了），然后再从头开始遍历直到set[x]里pos前的一个数，如果没有小于pos的数，那么我们只要遍历到set[x]的最后一个数就好了，这样就一次性把所有的x都扔掉了，同时last的位置也确定了，就可以进行x+1的处理。

主要思想是将问题分段化处理，使得每一段的处理过程都相同。

代码：

#include<bits/stdc++.h>using namespace std;#define ll long longconst int MAXN = 100010;typedef pair<int, int> P;int bit[MAXN];int p[MAXN];void add(int i, int x){    while(i < MAXN)    {        bit[i] +=x;        i += -i & i;    }}int sum(int i){    int sum = 0;    while(i)    {        sum += bit[i];        i -= -i & i;    }    return sum;}set<int>s[MAXN];set<int>::iterator it1, it2;int main(){    int n;    cin >> n;    for(int i=1;i<=n;i++)        scanf("%d",&p[i]), s[p[i]].insert(i), add(i, 1);    ll ans = 0, up, last = 0;    up = *max_element(p + 1, p + n + 1);    for(int i=1;i<=up;i++)    {        if(s[i].empty()) continue;        it2 = it1 = lower_bound(s[i].begin(), s[i].end(), last);        if(it2 != s[i].begin())        ans += sum(n) - sum(last);        else        ans += sum(*s[i].rbegin()) - sum(last);        while(it1 != s[i].end())        {            add(*it1, -1);last = *it1;            it1++;        }        if(it2 != s[i].begin())        {        for(it1=s[i].begin();it1!=it2;it1++)            {            ans++;            add(*it1, -1);            last = *it1;}ans += sum(last);}    }    cout << ans << endl;}