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