QOJ.ac

QOJ

ID题目提交者结果用时内存语言文件大小提交时间测评时间
#137715#6729. Unrooted TrieYarema#AC ✓264ms32792kbC++171.8kb2023-08-10 17:01:352023-08-10 17:01:36

Judging History

你现在查看的是最新测评结果

  • [2023-08-10 23:21:45]
  • System Update: QOJ starts to keep a history of the judgings of all the submissions.
  • [2023-08-10 17:01:36]
  • 评测
  • 测评结果:AC
  • 用时:264ms
  • 内存:32792kb
  • [2023-08-10 17:01:35]
  • 提交

answer

#include <bits/stdc++.h>
using namespace std;

#define SZ(a) (int)a.size()
#define ALL(a) a.begin(), a.end()
#define FOR(i, a, b) for (int i = (a); i<(b); ++i)
#define RFOR(i, b, a) for (int i = (b)-1; i>=(a); --i)
#define MP make_pair
#define PB push_back
#define F first
#define S second
#define FILL(a, b) memset(a, b, sizeof(a))

typedef long long LL;
typedef pair<int, int> PII;
typedef vector<int> VI;

const int N = 100447;
LL a[N];
LL ans[N];
vector<PII> g[N];

void dfs(int v, int par = -1)
{
	vector<int> cnt(26);
	for (auto [to, c] : g[v])
	{
		if (to != par)
			dfs(to, v);
		cnt[c]++;
	}
	int two = -1;
	FOR (i, 0, 26)
	{
		if (cnt[i] > 2)
		{ 
			a[0] = N;
			return;
		}
		if (cnt[i] == 2)
		{
			if (two == -1) two = i;
			else a[0] = N;
		}
	}
	if (two == -1) return;
	bool is_par = 0;
	for (auto [to, c] : g[v]) if (to == par && c == two) is_par = 1;
	if (is_par)
	{
		a[v]++;
		for (auto [to, c] : g[v]) if (c == two && to != par) a[to]--;
	}
	else
	{
		a[0]++;
		for (auto [to, c] : g[v]) if (c == two) a[to]--;
	}
}

void dfs2(int v, int par = -1, LL x = 0)
{
	x += a[v];
	ans[v] = x;
	for (auto [to, c] : g[v])
		if (to != par)
			dfs2(to, v, x);
}

void solve()
{
	int n;
	cin >> n;
	FOR (i, 0, n - 1)
	{
		int u, v;
		char ch;
		cin >> u >> v >> ch;
		u--, v--;
		g[u].PB({v, ch - 'a'});
		g[v].PB({u, ch - 'a'});
	}
	dfs(0);
	dfs2(0);
	int res = 0;
	FOR (i, 0, n) 
	{
		//cerr << i << ' ' << ans[i] << ' ' << a[i] << '\n';
		if (ans[i] == 0) 
			res++;
	}
		
	cout << res << '\n';
	FOR (i, 0, n) 
	{
		g[i].clear();
		a[i] = 0;
		ans[i] = 0;
	}
	
}

int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	
	int t;
	cin >> t;
	while (t--)
	{
		solve();
	}
	
	return 0;
}

这程序好像有点Bug,我给组数据试试?

详细

Test #1:

score: 100
Accepted
time: 1ms
memory: 5848kb

input:

2
6
3 1 a
3 2 a
3 4 b
4 5 c
4 6 d
6
3 1 a
3 2 a
3 4 b
5 4 c
6 4 c

output:

2
0

result:

ok 2 number(s): "2 0"

Test #2:

score: 0
Accepted
time: 264ms
memory: 32792kb

input:

1112
19
15 18 a
7 18 c
11 14 b
10 17 b
8 14 a
1 3 b
12 2 a
16 3 c
16 4 b
2 3 a
15 5 a
3 18 d
16 9 a
18 13 b
8 4 b
17 7 a
9 6 a
13 19 a
52
8 32 a
14 51 a
30 52 a
48 36 b
27 39 c
39 51 b
35 15 a
51 52 d
45 51 e
39 26 a
20 12 b
34 18 a
9 12 e
25 5 a
9 13 b
41 51 c
1 52 n
33 14 c
22 30 b
17 4 b
12 52 c
...

output:

15
49
22
68
34
17
28
27
3
4
34
70
37
39
19
24
58
8
16
14
10
73
73
65
35
45
33
81
46
35
78
49
22
13
26
10
33
48
47
3
9
50
8
37
15
84
23
75
26
35
35
61
65
58
30
56
11
8
39
60
88
40
56
17
42
62
12
11
2
59
22
54
14
91
87
1
80
11
45
69
80
33
87
46
56
62
54
80
7
4
48
20
55
19
9
4
38
39
89
35
63
46
24
7
52...

result:

ok 1112 numbers