#302764 (C) No.649 ここでちょっとQK！

提出ソース

結果

問題	No.649 ここでちょっとQK！
ユーザー	akakimidori
提出日時	2018-12-09 02:44:15
言語	C (gcc 12.3.0)
結果	WA
実行時間	-
コード長	7,396 bytes
コンパイル時間	708 ms
コンパイル使用メモリ	31,292 KB
実行使用メモリ	14,384 KB
最終ジャッジ日時	2023-10-13 11:02:27
合計ジャッジ時間	4,839 ms
ジャッジサーバーID （参考情報）	judge12 / judge15

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	1 ms 4,348 KB
testcase_01	AC	0 ms 4,348 KB
testcase_02	AC	1 ms 4,352 KB
testcase_03	AC	22 ms 4,348 KB
testcase_04	AC	118 ms 14,232 KB
testcase_05	AC	113 ms 14,384 KB
testcase_06	AC	114 ms 14,264 KB
testcase_07	AC	1 ms 4,352 KB
testcase_08	AC	1 ms 4,348 KB
testcase_09	WA	-
testcase_10	WA	-
testcase_11	WA	-
testcase_12	AC	63 ms 6,800 KB
testcase_13	AC	62 ms 6,688 KB
testcase_14	AC	60 ms 6,696 KB
testcase_15	AC	64 ms 6,896 KB
testcase_16	AC	63 ms 7,348 KB
testcase_17	AC	68 ms 7,828 KB
testcase_18	AC	78 ms 8,268 KB
testcase_19	AC	86 ms 8,692 KB
testcase_20	AC	97 ms 9,172 KB
testcase_21	AC	116 ms 9,792 KB
testcase_22	AC	119 ms 10,340 KB
testcase_23	AC	131 ms 10,600 KB
testcase_24	AC	137 ms 11,252 KB
testcase_25	AC	147 ms 11,664 KB
testcase_26	AC	159 ms 12,224 KB
testcase_27	AC	1 ms 4,352 KB
testcase_28	AC	1 ms 4,352 KB
testcase_29	AC	1 ms 4,352 KB
testcase_30	AC	55 ms 6,788 KB
testcase_31	AC	56 ms 7,048 KB
testcase_32	AC	1 ms 4,352 KB
testcase_33	AC	1 ms 4,352 KB
testcase_34	AC	1 ms 4,348 KB
testcase_35	AC	1 ms 4,348 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<assert.h>

typedef struct AVLNode {
  short rank;
  short bias;
  size_t size;
  struct AVLNode *parent;
  struct AVLNode *left;
  struct AVLNode *right;
  unsigned long long int val[];
} node;

node* newNode (const void *val, const size_t size) {
  node *r = (node *) malloc (sizeof (node) + size);
  r -> rank = 1;
  r -> bias = 0;
  r -> size = 1;
  r -> parent = NULL;
  r -> left = NULL;
  r -> right = NULL;
  memcpy (r -> val, val, size);
  return r;
}

short getRank (const node *t) {
  return t == NULL ? 0 : t -> rank;
}

short getBias (const node *t) {
  return t == NULL ? 0 : t -> bias;
}

int getSize (const node *t) {
  return t == NULL ? 0 : t -> size;
}

void update (node *t) {
  short lrank = getRank (t -> left);
  short rrank = getRank (t -> right);
  t -> rank = (lrank > rrank ? lrank : rrank) + 1;
  t -> bias = lrank - rrank;
  t -> size = getSize (t -> left) + getSize (t -> right) + 1;
}

int isLeftChild (const node *t) {
  return t != NULL && t -> parent != NULL && t -> parent -> left == t;
}

void concatL (node *parent, node *child) {
  if(child != NULL) {
    child -> parent = parent;
  }
  if(parent != NULL) {
    parent -> left = child;
  }
}

void concatR (node *parent, node *child) {
  if(child != NULL) {
    child -> parent = parent;
  }
  if(parent != NULL) {
    parent -> right = child;
  }
}

node* rightRotate (node *v) {
  assert(v !=NULL && v -> left != NULL);
  node *u = v -> left;
  concatL (v, u -> right);
  if (isLeftChild (v)) {
    concatL (v -> parent, u);
  } else {
    concatR (v -> parent, u);
  }
  concatR (u, v);
  update (v);
  update (u);
  return u;
}

node* leftRotate (node *u) {
  assert(u !=NULL && u -> right != NULL);
  node *v = u -> right;
  concatR (u, v -> left);
  if (isLeftChild (u)) {
    concatL (u -> parent, v);
  } else {
    concatR (u -> parent, v);
  }
  concatL (v, u);
  update (u);
  update (v);
  return v;
}

node* balance (node *t) {
  const short b = getBias (t);
  if (-1 <= b && b <= 1) {
    return t;
  }
  if (b < -1) {
    if (getBias (t -> right) >= 1) {
      rightRotate (t -> right);
    }
    return leftRotate (t);
  } else {
    if(getBias (t -> left) <= -1){
      leftRotate (t -> left);
    }
    return rightRotate (t);
  }
}

node* saveTree (node *t) {
  node *now = t;
  node *before = NULL;
  while (now != NULL) {
    update (now);
    before = balance (now);
    now = before -> parent;
  }
  return before;
}

typedef struct MultiSet_AVL {
  node *root;
  size_t valSize;
  int (*cmp) (const void *, const void *);
} multiset;

multiset* newMultiSet (const size_t size, int (*cmp) (const void *, const void *)) {
  multiset *mset = (multiset *)malloc(sizeof (multiset));
  mset -> root = NULL;
  mset -> valSize = size;
  mset -> cmp = cmp;
  return mset;
}

int getSetSize (const multiset *set) {
  return getSize (set -> root);
}

node* getMaxRef (const multiset *set) {
  if (set -> root == NULL) {
    return NULL;
  }
  node *now = set -> root;
  while (now -> right != NULL) {
    now = now -> right;
  }
  return now;
}

node* getMinRef (multiset *set) {
  if (set -> root == NULL) {
    return NULL;
  }
  node *now = set -> root;
  while (now -> left != NULL) {
    now = now -> left;
  }
  return now;
}

node* getAtRef (multiset *set, int k) {
  if (k <= 0 || getSetSize (set) < k) {
    return NULL;
  }
  node *now = set -> root;
  while (1) {
    int lSize = getSize (now -> left);
    if (k == lSize + 1){
      return now;
    }
    if (k <= lSize) {
      now = now -> left;
    } else {
      k -= lSize + 1;
      now = now -> right;
    }
  }
}

node* nextRef(node *t){
  if (t -> right != NULL) {
    node *s = t -> right;
    while (s -> left != NULL) {
      s = s-> left;
    }
    return s;
  }
  node *now = t -> parent;
  node *before = t;
  while (now != NULL && now -> left != before) {
    before = now;
    now = now -> parent;
  }
  return now;
}

node* prevRef(node *t){
  if (t -> left != NULL) {
    node *s = t -> left;
    while (s -> right != NULL) {
      s = s-> right;
    }
    return s;
  }
  node *now = t -> parent;
  node *before = t;
  while (now != NULL && now -> right != before) {
    before = now;
    now = now -> parent;
  }
  return now;
}

node* insert (multiset *set, const void *val) {
  node * const in = newNode (val, set -> valSize);
  if (set -> root == NULL) {
    set -> root = in;
    return in;
  }
  int (*cmp) (const void *, const void *) = set -> cmp;
  node *before = NULL;
  node *now = set -> root;
  int cmpRes = 0;
  while (now != NULL) {
    before = now;
    cmpRes = cmp (in -> val, now -> val);
    now = cmpRes <= 0 ? now -> left : now -> right;
  }
  if (cmpRes <= 0) {
    concatL (before, in);
  } else {
    concatR (before, in);
  }
  set -> root = saveTree (before);
  return in;
}

void eraseByRef (multiset * const set, node * const t) {
  if (t == NULL) {
    return;
  }
  if (t -> left == NULL) {
    if (isLeftChild (t)) {
      concatL (t -> parent, t -> right);
    } else {
      concatR (t -> parent, t -> right);
    }
    set -> root = saveTree (t -> parent);
    free (t);
  } else if (t -> right == NULL) {
    if (isLeftChild (t)) {
      concatL (t -> parent, t -> left);
    } else {
      concatR (t -> parent, t -> left);
    }
    set -> root = saveTree (t -> parent);
    free (t);
  } else {
    node *max = t -> left;
    while (max -> right != NULL) {
      max = max -> right;
    }
    node *p = NULL;
    if (max != t -> left) {
      concatR (max -> parent, max -> left);
      t -> left -> parent = NULL;
      p = saveTree (max -> parent);
    } else {
      p = t -> left -> left;
    }
    concatL (max, p);
    concatR (max, t -> right);
    if(isLeftChild (t)) {
      concatL (t -> parent, max);
    } else {
      concatR (t -> parent, max);
    }
    set -> root = saveTree (max);
    free (t);
  }
}

void eraseAt (multiset *set, const int k) {
  if (k < 1 || getSize (set -> root) < k) {
    return;
  }
  eraseByRef (set, getAtRef (set, k));
}

typedef long long int int64;

void print_rec(node *p){
  if(p==NULL){
    printf("Nil");
    return;
  }
  putchar('(');
  print_rec(p->left);
  putchar(',');
  printf("%lld",*(int64 *)p->val);
  putchar(',');
  print_rec(p->right);
  putchar(')');
}

void printSet(multiset *set){
  printf("set :");
  node *now = getMinRef(set);
  while(now!=NULL){
    printf(" %lld",*(int64 *)now->val);
    now=nextRef(now);
  }
  printf("\n");
  printf("rev :");
  now = getMaxRef(set);
  while(now!=NULL){
    printf(" %lld",*(int64 *)now->val);
    now=prevRef(now);
  }
  printf("\n");
  printf("rec :");
  print_rec(set->root);
  printf("\n");
  return;
}

int cmp(const void *a,const void *b){
  int64 p=*(int64 *)a;
  int64 q=*(int64 *)b;
  return p==q?0:p-q<0?-1:1;
}

void run(void){
  int q,k;
  scanf("%d%d",&q,&k);
  multiset *set=newMultiSet(sizeof(int64),cmp);
  while(q--){
    //assert(set->root==NULL || set->root->parent==NULL);
    //printf("size : %d\n",getSetSize(set));
    //printSet(set);
    int c;
    scanf("%d",&c);
    if(c==1){
      int64 v;
      scanf("%lld",&v);
      insert(set,&v);
    } else if(getSetSize(set)<k){
      printf("-1\n");
    } else {
      node *p=getAtRef(set,k);
      int64 v=*(int64 *)p->val;
      eraseByRef(set, p);
      printf("%lld\n",v);
    }
  }
}

int main(void){
  run();
  return 0;
}

yukicoder

結果

テストケース

ソースコード