Codeforces Round #424 (Div. 2) E. Cards Sorting(思维 树状数组)

来源:互联网 发布:服装设计用什么软件 编辑:程序博客网 时间:2024/06/06 14:15

题意:有n张牌,每张牌上有一个数,每次从牌顶取一张牌,如果这张牌是剩余牌中的最小值那么就把它扔掉,不然放到牌堆底,问

多少次操作才能把扔光。


思路:树状数组维护每个位置的牌是否存在,这样就能logn计算两个位置之间牌数,一共只有1e5个数,这样我们就可以从1开始枚

举模拟(一开始开一个1e5的vector把每个数的下标存进去),如果存在这个数,就从当前位置把所有的这个数拿完,记录维护下走

到的位置和操作步数,并将拿走的位置更新为0. 



代码:

#include<bits/stdc++.h>using namespace std;typedef long long ll;const int maxn = 1e5+5;int tree[maxn], a[maxn];vector<int> g[maxn];int lowbit(int x) { return x&(-x); }void update(int pos, int val){    while(pos < maxn)    {        tree[pos] += val;        pos += lowbit(pos);    }}int query(int pos){    int sum = 0;    while(pos)    {        sum += tree[pos];        pos -= lowbit(pos);    }    return sum;}int Find(int i, int p){    int l = 0, r = g[i].size()-1, ans = 0;    while(l <= r)    {        int mid = (l+r)/2;        if(g[i][mid] >= p) ans = mid, r = mid-1;        else l = mid+1;    }    return ans;}int dis(int l, int r){    if(l > r) return query(r)+query(maxn-1)-query(l);    else return query(r)-query(l);}int main(void){    int n;    while(cin >> n)    {        memset(tree, 0, sizeof(tree));        for(int i = 0; i < maxn; i++)            g[i].clear();        for(int i = 1; i <= n; i++)        {            scanf("%d", &a[i]);            g[a[i]].push_back(i);            update(i, 1);        }        int now = 0;        ll ans = 0;        for(int i = 1; i < maxn; i++)        {            if(g[i].size() == 0) continue;            int tmp = Find(i, now); //找到最近的下一个            int e = g[i].size();            //把i取完            for(int j = tmp; j < e; j++)            {                ans += dis(now, g[i][j]);                now = g[i][j];                update(now, -1);            }            for(int j = 0; j < tmp; j++)            {                ans += dis(now, g[i][j]);                now = g[i][j];                update(now, -1);            }        }        printf("%I64d\n", ans);    }    return 0;}


阅读全文
1 0