QOJ.ac
QOJ
ID | 题目 | 提交者 | 结果 | 用时 | 内存 | 语言 | 文件大小 | 提交时间 | 测评时间 |
---|---|---|---|---|---|---|---|---|---|
#641513 | #6729. Unrooted Trie | zhicheng | AC ✓ | 277ms | 24204kb | C++14 | 1.4kb | 2024-10-14 20:52:36 | 2024-10-14 20:52:37 |
Judging History
answer
#include<bits/stdc++.h>
using namespace std;
const int N=200010;
int cnt,f[N],siz[N],dfn[N],cntt[30],sum[N];
char s[3];
vector<pair<int,int> >v[N];
void add(int x,int y,int z){
v[x].push_back({y,z});
v[y].push_back({x,z});
}
void dfs(int u,int fa){
f[u]=fa;
siz[u]=1;
dfn[u]=++cnt;
for(auto i:v[u]){
if(i.first!=fa){
dfs(i.first,u);
siz[u]+=siz[i.first];
}
}
}
void upd(int x,int y){
sum[x]++;
sum[y+1]--;
}
int main(){
int t,n,a,b,pos,all=0,ans=0;
scanf("%d",&t);
while(t--){
all=ans=cnt=0;
scanf("%d",&n);
for(int i=1;i<=n;i++){
sum[i]=0;
v[i].clear();
}
for(int i=1;i<=n-1;i++){
scanf("%d%d%s",&a,&b,s);
add(a,b,s[0]-'a'+1);
}
dfs(1,0);
for(int i=1;i<=n;i++){
for(int j=1;j<=26;j++){
cntt[j]=0;
}
for(auto j:v[i]){
cntt[j.second]++;
}
pos=0;
for(int j=1;j<=26;j++){
if(cntt[j]>=3){
printf("0\n");
goto lass;
}
else if(cntt[j]==2){
if(pos){
printf("0\n");
goto lass;
}
pos=j;
}
}
if(!pos){
continue;
}
all++;
for(auto j:v[i]){
if(j.second==pos){
if(j.first==f[i]){
upd(1,dfn[i]-1);
upd(dfn[i]+siz[i],n);
}
else{
upd(dfn[j.first],dfn[j.first]+siz[j.first]-1);
}
}
}
}
for(int i=1;i<=n;i++){
sum[i]+=sum[i-1];
if(sum[i]==all){
ans++;
}
}
printf("%d\n",ans);
lass:;
}
}
这程序好像有点Bug,我给组数据试试?
详细
Test #1:
score: 100
Accepted
time: 2ms
memory: 9860kb
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: 277ms
memory: 24204kb
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