結果

問題 No.449 ゆきこーだーの雨と雪 (4)
ユーザー startcppstartcpp
提出日時 2016-11-19 00:53:27
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 402 ms / 5,000 ms
コード長 4,613 bytes
コンパイル時間 1,011 ms
コンパイル使用メモリ 79,636 KB
実行使用メモリ 31,344 KB
最終ジャッジ日時 2024-09-22 10:21:33
合計ジャッジ時間 13,766 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 5 ms
9,540 KB
testcase_01 AC 5 ms
9,536 KB
testcase_02 AC 4 ms
10,824 KB
testcase_03 AC 6 ms
9,796 KB
testcase_04 AC 6 ms
10,352 KB
testcase_05 AC 5 ms
9,940 KB
testcase_06 AC 5 ms
10,372 KB
testcase_07 AC 5 ms
9,796 KB
testcase_08 AC 5 ms
9,948 KB
testcase_09 AC 6 ms
9,984 KB
testcase_10 AC 5 ms
10,084 KB
testcase_11 AC 5 ms
10,396 KB
testcase_12 AC 239 ms
17,176 KB
testcase_13 AC 328 ms
18,940 KB
testcase_14 AC 402 ms
18,248 KB
testcase_15 AC 297 ms
19,844 KB
testcase_16 AC 365 ms
18,960 KB
testcase_17 AC 362 ms
18,432 KB
testcase_18 AC 239 ms
16,500 KB
testcase_19 AC 397 ms
18,812 KB
testcase_20 AC 394 ms
18,064 KB
testcase_21 AC 326 ms
18,868 KB
testcase_22 AC 332 ms
19,756 KB
testcase_23 AC 237 ms
17,272 KB
testcase_24 AC 371 ms
18,660 KB
testcase_25 AC 269 ms
15,792 KB
testcase_26 AC 299 ms
15,008 KB
testcase_27 AC 329 ms
18,964 KB
testcase_28 AC 335 ms
18,828 KB
testcase_29 AC 401 ms
18,632 KB
testcase_30 AC 292 ms
19,524 KB
testcase_31 AC 332 ms
19,768 KB
testcase_32 AC 389 ms
18,264 KB
testcase_33 AC 360 ms
18,228 KB
testcase_34 AC 272 ms
15,068 KB
testcase_35 AC 284 ms
19,396 KB
testcase_36 AC 271 ms
15,612 KB
testcase_37 AC 372 ms
31,320 KB
testcase_38 AC 356 ms
25,376 KB
testcase_39 AC 311 ms
24,420 KB
testcase_40 AC 315 ms
24,356 KB
testcase_41 AC 269 ms
27,164 KB
testcase_42 AC 270 ms
27,864 KB
testcase_43 AC 237 ms
21,736 KB
testcase_44 AC 161 ms
22,488 KB
testcase_45 AC 333 ms
31,344 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In member function ‘avl_tree<T>::node* avl_tree<T>::rank(avl_tree<T>::node*, long long int) const [with T = long long int]’:
main.cpp:83:3: warning: control reaches end of non-void function [-Wreturn-type]
   83 |   }
      |   ^

ソースコード

diff # 

//なんかAC数のカウントを忘れてた
#include <iostream>
#include <vector>
#include <string>
#include <algorithm>
#define int long long
using namespace std;

//http://www.prefield.com/algorithm/container/avl_tree.html
template <class T>
struct avl_tree {
  struct node {
    T key;
    int size, height;
    node *child[2];
    node(const T &key) : key(key), size(1), height(1) {
      child[0] = child[1] = 0; }
  } *root;
  typedef node *pointer;
  avl_tree() { root = NULL; }

  pointer find(const T &key) { return find(root, key); }
  node *find(node *t, const T &key) {
    if (t == NULL) return NULL;
    if (key == t->key) return t;
    else if (key < t->key) return find(t->child[0], key);
    else                   return find(t->child[1], key);
  }
  void insert(const T &key) { root = insert(root, new node(key)); }
  node *insert(node *t, node *x) {
    if (t == NULL) return x;
    if (x->key <= t->key) t->child[0] = insert(t->child[0], x);
    else                  t->child[1] = insert(t->child[1], x);
    t->size += 1;
    return balance(t);
  }
  void erase(const T &key) { root = erase(root, key); }
  node *erase(node *t, const T &x) {
    if (t == NULL) return NULL;
    if (x == t->key) {
      return move_down(t->child[0], t->child[1]);
    } else {
      if (x < t->key) t->child[0] = erase(t->child[0], x);
      else            t->child[1] = erase(t->child[1], x);
      t->size -= 1;
      return balance(t);
    }
  }
  node *move_down(node *t, node *rhs) {
    if (t == NULL) return rhs;
    t->child[1] = move_down(t->child[1], rhs);
    return balance(t);
  }
#define sz(t) (t ? t->size : 0)
#define ht(t) (t ? t->height : 0)
  node *rotate(node *t, int l, int r) {
    node *s = t->child[r];
    t->child[r] = s->child[l];
    s->child[l] = balance(t);
    if (t) t->size = sz(t->child[0]) + sz(t->child[1]) + 1;
    if (s) s->size = sz(s->child[0]) + sz(s->child[1]) + 1;
    return balance(s);
  }
  node *balance(node *t) {
    for (int i = 0; i < 2; ++i) {
      if (ht(t->child[!i]) - ht(t->child[i]) < -1) {
        if (ht(t->child[i]->child[!i]) - ht(t->child[i]->child[i]) > 0)
          t->child[i] = rotate(t->child[i], i, !i);
        return rotate(t, !i, i);
      }
    }
    if (t) t->height = max(ht(t->child[0]), ht(t->child[1])) + 1;
    if (t) t->size = sz(t->child[0]) + sz(t->child[1]) + 1;
    return t;
  }
  pointer rank(int k) const { return rank(root, k); }
  pointer rank(node *t, int k) const {
    if (!t) return NULL;
    int m = sz(t->child[0]);
    if (k  < m) return rank(t->child[0], k);
    if (k == m) return t;
    if (k  > m) return rank(t->child[1], k - m - 1);
  }
};

int n;
int level[26];
int t;
string name[100000], prob[100000];
vector<string> nameSorted;


void input() {
	cin >> n;
	for (int i = 0; i < n; i++) cin >> level[i];
	cin >> t;
	for (int i = 0; i < t; i++) cin >> name[i] >> prob[i];
	for (int i = 0; i < t; i++) nameSorted.push_back(name[i]);
	sort(nameSorted.begin(), nameSorted.end());
	nameSorted.erase(unique(nameSorted.begin(), nameSorted.end()), nameSorted.end());	//重複削除
}

//nameSortedを使用
int get_id(string name) {
	return lower_bound(nameSorted.begin(), nameSorted.end(), name) - nameSorted.begin();	//出席番号を返す
}

string get_name(int id) {
	return nameSorted[id];
}

int get_score(char prob, int ac_cnt) {
	int lv = level[prob - 'A'];
	return 50 * lv + 50 * lv / (0.8 + 0.2 * (1 + ac_cnt));
}

//集計
avl_tree<long long> bstree;	//-score[i] * t + last[i]を入れると, 高い順位ほど小さな値になるのでうれしい。

void syukei() {
	int i;
	int ac_cnt[26] = {0};
	static int score[100000] = {0};		//score[i] = 出席番号iの人の合計スコア
	static int last_ac[100000] = {0};	//last[i] = 出席番号iの人のラストac
	
	for (i = 0; i < nameSorted.size(); i++) {
		bstree.insert(1145141919893810);
	}
	
	for (i = 0; i < t; i++) {
		int id = get_id(name[i]);
		int prob_id = prob[i][0] - 'A';
		
		if (0 <= prob_id && prob_id < n) {
			if (score[id] == 0) { bstree.erase(1145141919893810); }
			else { bstree.erase(-score[id] * t + last_ac[id]); }
			
			score[id] += get_score(prob[i][0], ac_cnt[prob_id]);
			last_ac[id] = i;
			
			ac_cnt[prob_id]++;
			
			bstree.insert(-score[id] * t + last_ac[id]);
		}
		else {
			int st = 0, ed = nameSorted.size(), mid;	//[st, ed), oooxxx
			while (ed - st >= 2) {
				mid = (st + ed) / 2;
				
				if (bstree.rank(mid)->key > -score[id] * t + last_ac[id]) {
					ed = mid;
				}
				else {
					st = mid;
				}
			}
			cout << st + 1 << endl;
		}
	}
}

signed main() {
	input();
	syukei();
	return 0;
}
0