#644683 (C++14) No.1098 LCAs

提出ソース

結果

問題	No.1098 LCAs
ユーザー	shimarut
提出日時	2021-04-04 23:29:58
言語	C++14 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	526 ms / 2,000 ms
コード長	3,282 bytes
コンパイル時間	4,099 ms
コンパイル使用メモリ	181,304 KB
実行使用メモリ	28,168 KB
最終ジャッジ日時	2023-08-28 11:09:58
合計ジャッジ時間	13,016 ms
ジャッジサーバーID （参考情報）	judge11 / judge13

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	5 ms 9,840 KB
testcase_01	AC	5 ms 10,028 KB
testcase_02	AC	5 ms 10,036 KB
testcase_03	AC	6 ms 9,952 KB
testcase_04	AC	5 ms 10,004 KB
testcase_05	AC	5 ms 10,064 KB
testcase_06	AC	5 ms 9,944 KB
testcase_07	AC	5 ms 10,104 KB
testcase_08	AC	5 ms 10,004 KB
testcase_09	AC	5 ms 9,900 KB
testcase_10	AC	5 ms 10,016 KB
testcase_11	AC	6 ms 10,144 KB
testcase_12	AC	5 ms 9,896 KB
testcase_13	AC	7 ms 9,944 KB
testcase_14	AC	8 ms 9,892 KB
testcase_15	AC	7 ms 9,936 KB
testcase_16	AC	7 ms 9,896 KB
testcase_17	AC	7 ms 10,040 KB
testcase_18	AC	470 ms 16,404 KB
testcase_19	AC	471 ms 16,412 KB
testcase_20	AC	477 ms 16,608 KB
testcase_21	AC	477 ms 16,536 KB
testcase_22	AC	476 ms 16,648 KB
testcase_23	AC	426 ms 16,932 KB
testcase_24	AC	417 ms 17,024 KB
testcase_25	AC	413 ms 17,468 KB
testcase_26	AC	439 ms 17,204 KB
testcase_27	AC	431 ms 17,208 KB
testcase_28	AC	524 ms 27,208 KB
testcase_29	AC	515 ms 28,168 KB
testcase_30	AC	526 ms 27,112 KB

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

ソースコード

raw source code

#pragma region
#define _USE_MATH_DEFINES
#include <iostream>
#include <string>
#include <algorithm>
#include <cmath>
#include <cstdlib>
#include <vector>
#include <map>
#include <queue>
#include <stack>
#include <set>
#include <list>
#include <iomanip>
#include <cstdint>
#include <bitset>
#include <sstream>
#include <regex>
#include <fstream>
#include <array>
#include <atcoder/all>

using namespace atcoder;
using mint = modint1000000007;
//using mint = modint998244353;
//using mint = modint;

using namespace std;
typedef long long ll;
using vi = vector<int>;
using vvi = vector<vi>;
using vl = vector<ll>;
using vvl = vector<vl>;
using vmint = vector<mint>;
using vvmint = vector<vmint>;
using pint = pair<int, int>;
using pll = pair<ll, ll>;
//#define rep(i, s, e) for (int(i) = (s); (i) < (e); ++(i))
#define rep(i, e) for (int(i) = 0; (i) < (e); ++(i))
#define rrep(i, s) for (int(i) = (s) - 1; (i) >= 0; --(i))
#define all(x) x.begin(),x.end()
#define rall(x) x.rbegin(),x.rend()

#pragma region UnionFind
struct UnionFind
{
	vector<int> par;

	UnionFind(int n) : par(n, -1) {}
	void init(int n) { par.assign(n, -1); }

	int root(int x)
	{
		if (par[x] < 0) return x;
		else return par[x] = root(par[x]);
	}

	bool issame(int x, int y)
	{
		return root(x) == root(y);
	}

	bool merge(int x, int y)
	{
		x = root(x); y = root(y);
		if (x == y) return false;
		if (par[x] > par[y]) swap(x, y);
		par[x] += par[y];
		par[y] = x;
		return true;
	}

	int size(int x)
	{
		return -par[root(x)];
	}
};
#pragma endregion
#pragma region GCD
ll gcd(ll a, ll b)
{
	if (b == 0)return a;
	return gcd(b, a%b);
}
#pragma endregion
#pragma region LCM
ll lcm(ll a, ll b)
{
	return a / gcd(a, b) * b;
}
#pragma endregion
#pragma region chmin
template<class T> inline bool chmin(T& a, T b)
{
	if (a > b)
	{
		a = b;
		return true;
	}
	return false;
}
#pragma endregion
#pragma region chmax
template<class T> inline bool chmax(T& a, T b)
{
	if (a < b)
	{
		a = b;
		return true;
	}
	return false;
}
#pragma endregion
#pragma region グリッド内チェック
bool out(int x, int y, int h, int w)
{
	if (x < 0 || h <= x || y < 0 || w <= y)return true;
	else return false;
}
#pragma endregion
#pragma region Dijkstra
vl dijkstra(vector<vector<pair<int, ll>>> v, int s)
{
	ll INF = 1e18;
	int MAX = 1e6;
	vl res(MAX, INF);
	priority_queue<pair<ll, int>, vector<pair<ll, int>>, greater<pair<ll, int>>> q;
	q.push({ 0,s });
	while (!q.empty())
	{
		int now;
		ll d;
		tie(d, now) = q.top();
		q.pop();
		if (!chmin(res[now], d))continue;
		for (auto p : v[now])
		{
			int next;
			ll c;
			tie(next, c) = p;
			if (res[next] <= res[now] + c)continue;
			q.push({ res[now] + c,next });
		}
	}
	return res;
}
#pragma endregion
#pragma endregion

int n;
vvi v(200100);
vl dp(200100);
vi d(200100);

ll dfs(int now, int p, int depth)
{
	d[now] = depth;
	dp[now] = 1;
	for (int next : v[now])
	{
		if (next == p)continue;
		dp[now] += dfs(next, now, depth + 1);
	}
	return dp[now];
}

int main()
{
	cin >> n;
	rep(i, n - 1)
	{
		int a, b; cin >> a >> b;
		--a, --b;
		v[a].push_back(b);
		v[b].push_back(a);
	}
	dfs(0, -1, 1);
	rep(i, n)
	{
		ll res = 1;
		for (int x : v[i])
		{
			if (d[x] < d[i])continue;
			res += dp[x] * 2;
			res += dp[x] * (dp[i] - dp[x] - 1);
		}
		cout << res << endl;
	}
}

yukicoder

結果

テストケース

ソースコード