結果
| 問題 |
No.1637 Easy Tree Query
|
| コンテスト | |
| ユーザー |
|
| 提出日時 | 2021-08-06 21:33:18 |
| 言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
| 結果 |
AC
|
| 実行時間 | 248 ms / 2,000 ms |
| コード長 | 8,119 bytes |
| コンパイル時間 | 3,435 ms |
| コンパイル使用メモリ | 234,704 KB |
| 実行使用メモリ | 25,116 KB |
| 最終ジャッジ日時 | 2024-09-17 03:44:46 |
| 合計ジャッジ時間 | 10,411 ms |
|
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 2 |
| other | AC * 33 |
ソースコード
#ifndef HIDDEN_IN_VISUAL_STUDIO // 無意味.折りたたむのが目的.
// 警告の抑制
#define _CRT_SECURE_NO_WARNINGS
// 使えるライブラリの読み込み
#include <bits/stdc++.h>
#include <functional> // function
using namespace std;
// 型名の短縮
using ll = long long; // -2^63 ~ 2^63 = 9 * 10^18
using ull = unsigned long long; // 0 ~ 2^64 = 1.8 * 10^19
using uint = unsigned int; // 0 ~ 2^32 = 4 * 10^9
using pii = pair<int, int>;
using pll = pair<ll, ll>;
// 定数の定義
const double PI = 3.141592653589793238462643383279; // 円周率
const double DEG = PI / 180.0; // θ [deg] = θ * DEG [rad]
const vector<int> dx4 = { 1, 0, -1, 0 }; // 4 近傍
const vector<int> dy4 = { 0, 1, 0, -1 };
const vector<int> dx8 = { 1, 1, 0, -1, -1, -1, 0, 1 }; // 8 近傍
const vector<int> dy8 = { 0, 1, 1, 1, 0, -1, -1, -1 };
const ll INFL = (ll)1e18;
const int INF = (int)1e9;
const double EPS = 1e-10; // 許容誤差に応じて調整
// 汎用マクロの定義
#define all(a) (a).begin(), (a).end()
#define sz(x) ((int)(x).size())
#define rep(i, n) for(int i = 0, i##_len = ((int)(n)); i < i##_len; ++i) // 0 から n-1 まで昇順
#define repi(i, s, t) for(int i = (int)(s), i##_end = ((int)(t)); i <= i##_end; ++i) // s から t まで昇順
#define repir(i, s, t) for(int i = (int)(s), i##_end = ((int)(t)); i >= i##_end; --i) // s から t まで降順
#define repe(i, a) for(const auto& i : (a)) // a の全要素
#define repb(set, d) for(int set = 0; set < (1 << (int)(d)); ++set) // d ビット全探索(昇順)
#define repbm(mid, set, d) for(int mid = set; mid < (1 << (int)(d)); mid = (mid + 1) | set) // set を含む部分集合の全探索(昇順)
#define repbs(sub, set) for (int sub = set, bsub = 1; bsub > 0; bsub = sub, sub = (sub - 1) & set) // set の部分集合の全探索(降順)
#define repbc(set, k, d) for (int set = (1 << k) - 1, lb, nx; set < (1 << n); lb = set & -set, nx = set + lb, set = (((set & ~nx) / lb) >> 1) | nx) // 大きさ k の部分集合の全探索
#define repp(a) sort(all(a)); for(bool a##_perm = true; a##_perm; a##_perm = next_permutation(all(a))) // a の順列全て(昇順)
#define repit(it, a) for(auto it = (a).begin(); it != (a).end(); it++) // イテレータを回す(昇順)
#define Yes(b) if(b){cout << "Yes" << endl;}else{cout << "No" << endl;}
#define Tak(b) if(b){cout << "Takahashi" << endl;}else{cout << "Aoki" << endl;}
// 汎用関数の定義
inline ll pow(ll n, int k) { ll v = 1; rep(i, k) v *= n; return v; } // 工夫が必要なほど k が大きかったらどうせオーバーフローするからこれでいい
inline ll pow(int n, int k) { ll v = 1; rep(i, k) v *= (ll)n; return v; }
template <class T> inline bool chmax(T& M, const T& x) { if (M < x) { M = x; return true; } return false; } // 最大値を更新(更新されたら true を返す)
template <class T> inline bool chmin(T& m, const T& x) { if (m > x) { m = x; return true; } return false; } // 最小値を更新(更新されたら true を返す)
// 入出力用の >>, << のオーバーロード
template <class T, class U> istream& operator>> (istream& is, pair<T, U>& p) { is >> p.first >> p.second; return is; } // pair の入力用
template <class T, class U> ostream& operator<< (ostream& os, const pair<T, U>& p) { os << "(" << p.first << "," << p.second << ")"; return os; } // pair の出力用
template <class T, class U, class V> istream& operator>> (istream& is, tuple<T, U, V>& t) { is >> get<0>(t) >> get<1>(t) >> get<2>(t); return is; } // tuple の入力用
template <class T, class U, class V> ostream& operator<< (ostream& os, const tuple<T, U, V>& t) { os << "(" << get<0>(t) << "," << get<1>(t) << "," << get<2>(t) << ")"; return os; } // tuple の出力用
template <class T, class U, class V, class W> istream& operator>> (istream& is, tuple<T, U, V, W>& t) { is >> get<0>(t) >> get<1>(t) >> get<2>(t) >> get<3>(t); return is; } // tuple の入力用
template <class T, class U, class V, class W> ostream& operator<< (ostream& os, const tuple<T, U, V, W>& t) { os << "(" << get<0>(t) << "," << get<1>(t) << "," << get<2>(t) << "," << get<3>(t) << ")"; return os; } // tuple の出力用
template <class T> istream& operator>> (istream& is, vector<T>& v) { rep(i, sz(v)) is >> v[i]; return is; } // vector の入力用
template <class T> ostream& operator<< (ostream& os, const vector<T>& v) { rep(i, sz(v)) os << v[i] << " "; return os; } // vector の出力用
template <class T> ostream& operator<< (ostream& os, const set<T>& s) { repe(x, s) os << x << " "; return os; } // set の出力用
// 手元環境(Visual Studio)
#ifdef _MSC_VER
#define popcount (int)__popcnt // 全ビットにおける 1 の個数
#define popcountll (int)__popcnt64
inline int ctz(uint n) { unsigned long i; _BitScanForward(&i, n); return i; } // 下位ビットに並ぶ 0 の個数
ll gcd(ll a, ll b) { return b ? gcd(b, a % b) : a; } // 最大公約数
#define dump(x) cerr << "[DEBUG] " << endl << x << endl; // デバッグ出力用
#define dumpel(v) cerr << "[DEBUG]" << endl; for (const auto& x : v) {cout << x << endl;}
// 提出用(GCC)
#else
#define popcount (int)__builtin_popcount
#define popcountll (int)__builtin_popcountll
#define ctz __builtin_ctz
#define gcd __gcd
#define dump(x)
#define dumpel(v)
#endif
#endif // 無意味.折りたたむのが目的.
// AtCoder 専用
#include <atcoder/all>
using namespace atcoder;
// mint で使いたい法によってここを切り替える
using mint = modint1000000007;
//using mint = modint998244353;
//using mint = modint; // modint::set_mod(10000); // mint の法の指定
istream& operator>> (istream& is, mint& x) { ll tmp; is >> tmp; x = tmp; return is; } // mint の入力用
ostream& operator<< (ostream& os, const mint& x) { os << x.val(); return os; } // mint の出力用
//【根付き木のノード】
struct TNode {
int parent = -1; // 親(なければ -1)
vector<int> child; // 子(なければ空リスト)
int depth = -1; // 深さ(根からのパスの長さ)
int height = -1; // 高さ(最も遠い葉へのパスの長さ)
// 出力
friend ostream& operator<<(ostream& os, const TNode& v) {
os << '(' << v.parent << ',' << v.child << ',' << v.depth << ',' << v.height << ')';
return os;
}
};
//【木 → 根付き木】O(|V|)
/*
* 木 g を根を r とする根付き木 rt として再構築する.
* 親,子,深さ,高さ(改変)の情報を正しく定める.
*/
void tree_to_rooted_tree(vector<vector<int>>& g, int r, vector<TNode>& rt) {
int n = (int)g.size();
rt = vector<TNode>(n); // 根付き木
// 再帰用の関数
// s : 注目ノード,p : s の親,d : s の深さ,戻り値:s の高さ
function<int(int, int, int)> dfs = [&](int s, int p, int d) {
// 行きがけに s の親と深さを定める.
rt[s].parent = p;
rt[s].depth = d;
int h = 0;
repe(t, g[s]) {
if (t == p) {
continue;
}
// 通りがけに s の子を定める.
rt[s].child.push_back(t);
h += dfs(t, s, d + 1);
}
// 帰りがけに高さを定める.
rt[s].height = h + 1;
return rt[s].height;
};
// 根 r を始点として再帰関数を呼び出す.
dfs(r, -1, 0);
}
//【無向グラフの構築】O(|E|)
/*
* 入力を受け取り n 頂点 m 辺の無向グラフを構成する.
* 入力の頂点番号は 1-indexed とする.
* g[i] : 頂点 i と辺でつながっている頂点のリスト
*/
void create_undirected_graph(int n, int m, vector<vector<int>>& g) {
g = vector<vector<int>>(n);
rep(i, m) {
int a, b;
cin >> a >> b;
a--;
b--;
g[a].push_back(b);
g[b].push_back(a);
}
}
int main() {
cout << fixed << setprecision(15); // 小数点以下の桁数の指定
int n, q;
cin >> n >> q;
vector<vector<int>> g;
create_undirected_graph(n, n - 1, g);
vector<TNode> rt;
tree_to_rooted_tree(g, 0, rt);
dumpel(rt);
ll res = 0;
rep(i, q) {
int p;
ll x;
cin >> p >> x;
p--;
res += (ll)rt[p].height * x;
cout << res << endl;
}
}