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

题意:有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;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值