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 long
const 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;
}