zoukankan      html  css  js  c++  java
  • Codeforces Round #424 (Div. 2, rated, based on VK Cup Finals) E. Cards Sorting 树状数组

    E. Cards Sorting
    time limit per test
    1 second
    memory limit per test
    256 megabytes
    input
    standard input
    output
    standard output

    Vasily has a deck of cards consisting of n cards. There is an integer on each of the cards, this integer is between 1 and 100 000, inclusive. It is possible that some cards have the same integers on them.

    Vasily decided to sort the cards. To do this, he repeatedly takes the top card from the deck, and if the number on it equals the minimum number written on the cards in the deck, then he places the card away. Otherwise, he puts it under the deck and takes the next card from the top, and so on. The process ends as soon as there are no cards in the deck. You can assume that Vasily always knows the minimum number written on some card in the remaining deck, but doesn't know where this card (or these cards) is.

    You are to determine the total number of times Vasily takes the top card from the deck.

    Input

    The first line contains single integer n (1 ≤ n ≤ 100 000) — the number of cards in the deck.

    The second line contains a sequence of n integers a1, a2, ..., an (1 ≤ ai ≤ 100 000), where ai is the number written on the i-th from top card in the deck.

    Output

    Print the total number of times Vasily takes the top card from the deck.

    Examples
    input
    4
    6 3 1 2
    output
    7
    input
    1
    1000
    output
    1
    input
    7
    3 3 3 3 3 3 3
    output
    7
    Note

    In the first example Vasily at first looks at the card with number 6 on it, puts it under the deck, then on the card with number 3, puts it under the deck, and then on the card with number 1. He places away the card with 1, because the number written on it is the minimum among the remaining cards. After that the cards from top to bottom are [2, 6, 3]. Then Vasily looks at the top card with number 2 and puts it away. After that the cards from top to bottom are [6, 3]. Then Vasily looks at card 6, puts it under the deck, then at card 3 and puts it away. Then there is only one card with number 6 on it, and Vasily looks at it and puts it away. Thus, in total Vasily looks at 7 cards.

    题意:n张牌,如果当前不是最小值,将其放到最底下,否则拿掉,问最少需要多少步;

    思路:就是模拟,然后对于求答案的时候需要更新,对于未拿走的牌,标记为1,否则为0,利用树状数组快速求区间和,即求到下张牌的距离;

    #pragma comment(linker, "/STACK:1024000000,1024000000")
    #include<iostream>
    #include<cstdio>
    #include<cmath>
    #include<string>
    #include<queue>
    #include<algorithm>
    #include<stack>
    #include<cstring>
    #include<vector>
    #include<list>
    #include<set>
    #include<map>
    using namespace std;
    #define LL long long
    #define pi (4*atan(1.0))
    #define eps 1e-14
    #define bug(x)  cout<<"bug"<<x<<endl;
    const int N=1e5+10,M=2e6+10,inf=1e9+10;
    const LL INF=1e18+10,mod=1e9+7;
    
    struct AYT
    {
        int tree[N];
        int lowbit(int x)
        {
            return x&-x;
        }
        void update(int x,int c)
        {
            while(x<N)
            {
                tree[x]+=c;
                x+=lowbit(x);
            }
        }
        int query(int x)
        {
            int ans=0;
            while(x)
            {
                ans+=tree[x];
                x-=lowbit(x);
            }
            return ans;
        }
    } tree;
    
    int a[N],n;
    vector<int>pos[N];
    int check(int i,int x)
    {
        int s=0,e=pos[i].size()-1,ans=-1;
        while(s<=e)
        {
            int mid=(s+e)>>1;
            if(pos[i][mid]>x)
            {
                ans=mid;
                e=mid-1;
            }
            else s=mid+1;
        }
        return ans;
    }
    int dis(int s,int e)
    {
        if(s==-1)return tree.query(e);
        if(s<e)return tree.query(e)-tree.query(s);
        return tree.query(n)-tree.query(s)+tree.query(e);
    }
    int main()
    {
        scanf("%d",&n);
        for(int i=1; i<=n; i++)
        {
            scanf("%d",&a[i]);
            tree.update(i,1);
            pos[a[i]].push_back(i);
        }
        int pre=-1;
        LL ans=0;
        for(int i=1; i<=100000; i++)
        {
            if(!pos[i].size())continue;
            int v=check(i,pre);
            for(int j=max(0,v); j<pos[i].size(); j++)
            {
                ans+=dis(pre,pos[i][j]);
                pre=pos[i][j];
                tree.update(pre,-1);
                //cout<<pre<<" "<<ans<<endl;
            }
            for(int j=0; j<v; j++)
            {
                ans+=dis(pre,pos[i][j]);
                pre=pos[i][j];
                tree.update(pre,-1);
                //cout<<pre<<" "<<ans<<endl;
            }
    
        }
        printf("%lld
    ",ans);
        return 0;
    }
  • 相关阅读:
    React Hooks 全解(一)
    Google搜索技巧
    #!/usr/bin/python3 和 #!/usr/bin/env python3 的区别
    Python函数
    Python程序代码阅读
    画个爱心向你表白
    自学需要注意的点
    Python文件操作
    国内加速访问GitHub
    (九) -前端-异步编程
  • 原文地址:https://www.cnblogs.com/jhz033/p/7203078.html
Copyright © 2011-2022 走看看