結果

問題 No.449 ゆきこーだーの雨と雪 (4)
ユーザー anta
提出日時 2016-11-18 06:11:49
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 102 ms / 5,000 ms
コード長 2,534 bytes
コンパイル時間 1,816 ms
コンパイル使用メモリ 183,612 KB
実行使用メモリ 16,640 KB
最終ジャッジ日時 2024-09-22 09:21:38
合計ジャッジ時間 6,115 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 43
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include "bits/stdc++.h"
using namespace std;
//Ω(同じスコアの人たち)かけるがビット並列バージョン
//おそらくAC

#pragma GCC optimize ("O3")
#pragma GCC target ("avx")

struct FenwickTree {
	typedef int T;
	vector<T> v;

	void init(int n) { v.assign(n, T()); }

	int size() const { return (int)v.size(); }

	void add(int i, T x) {
		for(; i < size(); i |= i + 1) v[i] += x;
	}

	T sum(int i) const {
		T r = T();
		for(-- i; i >= 0; i = (i & (i + 1)) - 1) r += v[i];
		return r;
	}
};

struct BitVector {
	std::vector<uint64_t> bits;
	int length;

	BitVector() : length(0) {}

	int size() const { return length; }

	void append1() {
		if((length ++ & 63) == 0)
			bits.push_back(1);
		else
			bits.back() |= uint64_t(1) << ((length - 1) & 63);
	}

	void reset(int i) {
		bits[i >> 6] &= ~(uint64_t(1) << (i & 63));
	}

	int sum(int i) const {
		int res = 0;
		int blocks = i >> 6;
		if(i & 63)
			res += popcount(bits[blocks] << (-i & 63));
		for(int k = blocks - 1; k >= 0; -- k)
			res += popcount(bits[k]);
		return res;
	}

	static int popcount(uint64_t x) {
#ifdef __GNUC__
		return __builtin_popcountll(x);
#else
		return (int)_mm_popcnt_u64(x);
#endif
	}
};

int main() {
	int N;
	scanf("%d", &N);
	vector<int> L(N);
	for(int i = 0; i < N; ++ i)
		scanf("%d", &L[i]);
	int T;
	scanf("%d", &T);

	struct Contestant {
		int totalScore;
		int lastTime;
		int subIndex;

		Contestant() :
			totalScore(0), lastTime(0),
			subIndex(-1) {}
	};

	std::map<std::string, Contestant> contestants;
	std::vector<int> count(N, 0);

	const int MaxScore = 15600;
	FenwickTree mainFT;
	mainFT.init(MaxScore + 1);
	vector<BitVector> bitVectors(MaxScore + 1);
	std::vector<int> scoreCount(MaxScore + 1);

	for(int i = 0; i < T; ++ i) {
		char name[17], P[2];
		scanf("%s%s", name, P);
		if(*P != '?') {
			int p = *P - 'A';

			int stars = L[p];
			int rank = ++ count[p];
			int score = 50 * stars + 50 * stars * 10 / (8 + 2 * rank);

			auto &c = contestants[name];

			if(c.totalScore != 0) {
				mainFT.add(c.totalScore, -1);
				bitVectors[c.totalScore].reset(c.subIndex);
			}

			c.totalScore += score;
			c.lastTime = i + 1;

			assert(c.totalScore <= MaxScore);
			mainFT.add(c.totalScore, 1);
			c.subIndex = bitVectors[c.totalScore].size();
			bitVectors[c.totalScore].append1();
		} else {
			auto &c = contestants[name];

			int ans = 0;
			ans += (int)contestants.size() - mainFT.sum(c.totalScore + 1);
			ans += bitVectors[c.totalScore].sum(c.subIndex);

			printf("%d\n", ans + 1);
		}
	}
	return 0;
}
0