結果

問題 No.778 クリスマスツリー
コンテスト
ユーザー i_mrhj
提出日時 2019-08-07 12:01:54
言語 C
(gcc 13.3.0)
結果
WA  
実行時間 -
コード長 1,918 bytes
コンパイル時間 306 ms
コンパイル使用メモリ 33,024 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-07-18 18:15:27
合計ジャッジ時間 3,411 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 WA * 1
other WA * 3 RE * 9
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <stdio.h>
#include <stdlib.h>
#include <stdbool.h>
#include <string.h>
#include <math.h>
#include <limits.h>
#define Max(a, b) ((a) > (b) ? (a) : (b))
#define Min(a, b) ((a) > (b) ? (b) : (a))
#define abs(x) ((x) > 0 ? (x) : -(x))
#define rep(i, n) for(int i = 0; i < (n); i++)
#define INF 1000000000000 //10^12
#define MOD 1000000007 //10^9 + 7
#define endl printf("\n")
typedef long long ll;

#define MAX_N  200010

int n, dat[2 * MAX_N - 1];

void
init(int n_)
{
  n = 1;
  while (n < n_) {
    n *= 2;
  }

  for (int i = 0; i < 2 * n - 1; i++) {
    dat[i] = 0;
  }
}

void
update(int k, int a)
{
  k += n - 1;
  dat[k] += a;

  while (k > 0) {
    k = (k - 1) / 2;
    dat[k] = dat[2 * k + 1] + dat[2 * k + 2];
  }
}

//[a, b)の和を求める
int
query(int a, int b, int k, int l, int r)
{
  if (r <= a || b <= l) {
    return 0;
  }

  if (a <= l && r <= b) {
    return dat[k];
  } else {
    int vl = query(a, b, k * 2 + 1, l, (l + r) / 2);
    int vr = query(a, b, k * 2 + 2, (l + r) / 2, r);
    return vl + vr;
  }
}

typedef struct { int u, v; } E;

int
asort(const void *a, const void *b)
{
  E x = *(E *)a, y = *(E *)b;
  if (x.u > y.u) return  1;
  if (x.u < y.u) return -1;
  return 0;
}

ll cnt = 0;

void
dfs(int i, E *e, int *id)
{
  update(i, 1);
  cnt += (ll)query(0, i, 0, 0, n);
  for (int j = id[i]; j < id[i + 1]; j++) {
    if (dat[n - 1 + e[j].v] == 0) {
      dfs(e[j].v, e, id);
    }
  }
  update(i, -1);
  return;
}
  
int
main(int argc, char **argv)
{
  int m;
  scanf("%d", &m);

  E e[200000];
  int a; 
  for (int i = 1; i < m; i++) {
    scanf("%d", &a);
    e[i - 1].u = a;
    e[i - 1].v = i;
  }

  qsort(e, m - 1, sizeof(int), asort);

  int id[200000];
  id[0] = 0;

  int p = 0;
  for (int i = 0; i < m; i++){
    while(e[p].u < i && p < m - 1) p++;
    id[i] = p;
  }

  id[m] = m - 1;

  init(m);
  dfs(0, e, id);

  printf("%lld\n", cnt);

  return 0;
}
0