kmjp's blog

競技プログラミング参加記です

Codeforces #424 Div1 B. Cards Sorting

ここら辺もまだ簡単だね。
http://codeforces.com/contest/830/problem/B

問題

N個のカードが山になっている。
各カードには1つの整数が書かれており、その値は全部わかっているものとする。
このカードに対し、山が空になるまで以下の処理を繰り返す。

  • 山の最上位のカードを見る。そこの数値が山の中で最小の場合、それを取り除く。
  • そうでない場合、カードを一番下に移動する。

解法

まずカードを取り除く順を求めよう。
これはカードを整数順にソートすれば容易である。
あとはBIT等使い、前に取り除いたカードのあと次に取り除くカードの間に何枚残っているカード(=下に動かさないといけないカードがあるか)を適宜数えていけばよい。

int N;
deque<int> D[101010];
vector<int> V;

template<class V, int ME> class BIT {
public:
	V bit[1<<ME];
	V operator()(int e) {V s=0;e++;while(e) s+=bit[e-1],e-=e&-e; return s;}
	V add(int e,V v) { e++; while(e<=1<<ME) bit[e-1]+=v,e+=e&-e;}
};
BIT<int,20> bt;

void solve() {
	int i,j,k,l,r,x,y; string s;
	
	cin>>N;
	FOR(i,N) {
		cin>>x;
		D[x].push_back(i+1);
		bt.add(i+1,1);
	}
	
	int pre=-1;
	FOR(x,101000) if(D[x].size()) {
		if(D[x].back()>pre) {
			while(D[x].front()<pre) D[x].push_back(D[x].front()), D[x].pop_front();
		}
		FORR(r,D[x]) {
			V.push_back(r);
			pre=r;
		}
	}
	
	pre=0;
	ll ret=0;
	FORR(r,V) {
		if(r>pre) {
			ret+=bt(r)-bt(pre);
		}
		else {
			ret+=bt(N)-bt(pre) + bt(r);
		}
		bt.add(r,-1);
		pre=r;
	}
	cout<<ret<<endl;
}

まとめ

750ptでもいい気がする?