結果

問題 No.1038 TreeAddQuery
ユーザー ei1333333ei1333333
提出日時 2020-03-15 23:46:53
言語 C#(csc)
(csc 3.9.0)
結果
AC  
実行時間 2,618 ms / 4,000 ms
コード長 3,900 bytes
コンパイル時間 4,654 ms
コンパイル使用メモリ 105,852 KB
実行使用メモリ 68,744 KB
最終ジャッジ日時 2023-08-16 14:53:32
合計ジャッジ時間 39,940 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 62 ms
22,796 KB
testcase_01 AC 59 ms
20,788 KB
testcase_02 AC 56 ms
20,692 KB
testcase_03 AC 86 ms
22,828 KB
testcase_04 AC 88 ms
20,816 KB
testcase_05 AC 82 ms
22,680 KB
testcase_06 AC 84 ms
20,684 KB
testcase_07 AC 89 ms
24,788 KB
testcase_08 AC 1,719 ms
46,804 KB
testcase_09 AC 1,847 ms
48,736 KB
testcase_10 AC 1,954 ms
48,800 KB
testcase_11 AC 1,962 ms
48,712 KB
testcase_12 AC 2,015 ms
50,904 KB
testcase_13 AC 2,618 ms
66,736 KB
testcase_14 AC 2,409 ms
53,516 KB
testcase_15 AC 2,339 ms
49,656 KB
testcase_16 AC 2,316 ms
50,540 KB
testcase_17 AC 2,333 ms
48,100 KB
testcase_18 AC 958 ms
49,408 KB
testcase_19 AC 1,077 ms
47,596 KB
testcase_20 AC 1,053 ms
49,244 KB
testcase_21 AC 1,107 ms
51,416 KB
testcase_22 AC 1,236 ms
49,236 KB
testcase_23 AC 1,290 ms
49,076 KB
testcase_24 AC 1,689 ms
46,684 KB
testcase_25 AC 2,562 ms
68,744 KB
testcase_26 AC 1,800 ms
60,544 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
Microsoft (R) Visual C# Compiler version 3.9.0-6.21124.20 (db94f4cc)
Copyright (C) Microsoft Corporation. All rights reserved.

ソースコード

diff # 

using System;
using System.Collections.Generic;

class Program
{
  Scanner cin;

  int N, Q;
  int[] A, B, C;
  List<int>[] g, qs;
  int[] sub, dep;
  long[] ans;
  bool[] used;

  BinaryIndexedTree bit;
  List<int> prop;


  void myon()
  {
    cin = new Scanner();

    N = cin.nextInt();
    Q = cin.nextInt();

    g = new List<int>[N];
    qs = new List<int>[N];
    sub = new int[N];
    dep = new int[N];
    used = new bool[N];

    A = new int[Q];
    B = new int[Q];
    C = new int[Q];
    ans = new long[Q];

    prop = new List<int>();
    
    for (int i = 0; i < N; i++)
    {
      g[i] = new List<int>();
      qs[i] = new List<int>();
    }

    for (int i = 1; i < N; i++)
    {
      int a = cin.nextInt() - 1;
      int b = cin.nextInt() - 1;
      g[a].Add(b);
      g[b].Add(a);
    }

    for (int i = 0; i < Q; i++)
    {
      A[i] = cin.nextInt() - 1;
      B[i] = cin.nextInt();
      C[i] = cin.nextInt();
      qs[A[i]].Add(i);
    }

    build(0);

    foreach (var i in ans)
    {
      Console.WriteLine(i);
    }
  }

  int build_dfs(int idx, int par)
  {
    sub[idx] = 1;
    foreach (var to in g[idx])
    {
      if (to == par || used[to]) continue;
      sub[idx] += build_dfs(to, idx);
    }
    return sub[idx];
  }

  int search_centroid(int idx, int par, int mid)
  {
    foreach (var to in g[idx])
    {
      if (to == par || used[to]) continue;
      if (sub[to] > mid) return search_centroid(to, idx, mid);
    }
    return idx;
  }

  int add_dfs(int idx, int par, int d)
  {
    dep[idx] = d;
    int max_d = d;
    foreach (var q in qs[idx])
    {
      prop.Add(q);
    }
    foreach (var to in g[idx])
    {
      if (to == par || used[to]) continue;
      max_d = Math.Max(max_d, add_dfs(to, idx, d + 1));
    }
    return max_d;
  }

  void build(int idx)
  { 
    int centroid = search_centroid(idx, -1, build_dfs(idx, -1) / 2);
    used[centroid] = true;

    int d = add_dfs(centroid, -1, 0);
    bit = new BinaryIndexedTree(d + 1);

    prop.Sort();
    foreach (var q in prop)
    {
      ans[q] += bit.query(dep[A[q]]);
      int rest = B[q] - dep[A[q]];
      if (rest >= 0)
      {
        bit.add(0, C[q]);
        bit.add(rest + 1, -C[q]);
      }
    }
    prop.Clear();

    foreach(var to in g[centroid])
    {
      if (used[to]) continue;
      d = add_dfs(to, centroid, 1);
      bit = new BinaryIndexedTree(d + 1);
      prop.Sort();
      foreach (var q in prop)
      {
        ans[q] -= bit.query(dep[A[q]]);
        int rest = B[q] - dep[A[q]];
        if (rest >= 0)
        {
          bit.add(0, C[q]);
          bit.add(rest + 1, -C[q]);
        }
      }
      prop.Clear();
    }

    foreach (var to in g[centroid])
    {
      if (used[to]) continue;
      build(to);
    }
  }

  static void Main(string[] args)
  {
    new Program().myon();
  }
}

class Scanner
{
  string[] s;
  int i;
  readonly char[] cs = new char[] { ' ' };

  public Scanner()
  {
    s = new string[0];
    i = 0;
  }

  public string next()
  {
    if (i < s.Length) return s[i++];
    string st = Console.ReadLine();
    while (st == "") st = Console.ReadLine();
    s = st.Split(cs, StringSplitOptions.RemoveEmptyEntries);
    if (s.Length == 0) return next();
    i = 0;
    return s[i++];
  }

  public string nextLine()
  {
    return Console.ReadLine();
  }

  public int nextInt()
  {
    return int.Parse(next());
  }

  public long nextLong()
  {
    return long.Parse(next());
  }

  public double nextDouble()
  {
    return double.Parse(next());
  }
}


class BinaryIndexedTree
{
  long[] data;

  public BinaryIndexedTree(int sz)
  {
    data = new long[sz + 1];
  }

  public void add(int k, long x)
  {
    for (++k; k < data.Length; k += k & -k)
    {
      data[k] += x;
    }
  }

  public long query(int k)
  {
    long ret = 0;
    for (++k; k > 0; k -= k & -k)
    {
      ret += data[k];
    }
    return ret;
  }
}
0