結果

問題 No.1103 Directed Length Sum
ユーザー takumattakumat
提出日時 2020-07-04 19:55:40
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 1,519 bytes
コンパイル時間 1,125 ms
コンパイル使用メモリ 111,356 KB
実行使用メモリ 151,916 KB
最終ジャッジ日時 2024-09-19 08:59:43
合計ジャッジ時間 21,002 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 1 ms
5,376 KB
testcase_02 WA -
testcase_03 AC 871 ms
73,600 KB
testcase_04 AC 1,154 ms
43,008 KB
testcase_05 AC 2,348 ms
71,936 KB
testcase_06 AC 690 ms
29,184 KB
testcase_07 AC 111 ms
9,600 KB
testcase_08 AC 186 ms
12,972 KB
testcase_09 AC 61 ms
7,296 KB
testcase_10 AC 270 ms
16,384 KB
testcase_11 AC 1,351 ms
46,688 KB
testcase_12 AC 704 ms
29,440 KB
testcase_13 AC 279 ms
16,768 KB
testcase_14 AC 46 ms
6,400 KB
testcase_15 AC 511 ms
23,680 KB
testcase_16 AC 1,485 ms
52,096 KB
testcase_17 AC 1,617 ms
54,016 KB
testcase_18 AC 286 ms
16,384 KB
testcase_19 AC 1,393 ms
47,836 KB
testcase_20 AC 84 ms
8,320 KB
testcase_21 AC 166 ms
11,904 KB
testcase_22 AC 1,034 ms
39,680 KB
testcase_23 AC 538 ms
25,216 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <stdio.h>
#include <string.h>
#include <limits.h>
#include <math.h>
#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
#include <map>
#include <set>
#include <stack>
#include <queue>

typedef unsigned long long ULLONG;
typedef long long LLONG;
static const LLONG MOD_NUM = 1000000007;//998244353;

template<class _T> static void getval(_T& a) {
	std::cin >> a;
}
template<class _T> static void getval(_T& a, _T& b) {
	std::cin >> a >> b;
}
template<class _T> static void getval(_T& a, _T& b, _T& c) {
	std::cin >> a >> b >> c;
}

static void exec();

int main()
{
	exec();
	fflush(stdout);
	return 0;
}

static LLONG dfs(std::vector<std::vector<int>>& graph, int here, int pass)
{
	LLONG ans = 0;
	int size = 0;
	if ((size = graph[here].size()) == 0) {
		ans = (pass * (pass + 1) / 2LL) % MOD_NUM;
	}
	else {
		for (auto it = graph[here].begin(); it != graph[here].end(); it++) {
			ans = (ans + dfs(graph, *it, pass + 1)) % MOD_NUM;
		}
		ans += (pass * (pass + 1) / 2LL) % MOD_NUM;
		ans %= MOD_NUM;
	}
	return ans;
}

static void exec()
{
	int N;
	getval(N);

	std::set<int> vers;
	for (int i = 0; i < N; i++) {
		vers.insert(i);
	}
	std::vector<std::vector<int>> graph(N);
	for (int i = 1; i < N; i++) {
		int a, b;
		getval(a, b); a--; b--;
		graph[a].push_back(b);
		vers.erase(b);
	}

	int root = *(vers.begin());
	LLONG ans = 0;
	for (auto it = graph[root].begin(); it != graph[root].end(); it++) {
		ans += dfs(graph, *it, 1);
		ans %= MOD_NUM;
	}
	printf("%lld\n", ans);
}
0