【题目描述】
给定 n 个长度不超过 10 的数字串,问其中是否存在两个数字串 S,T,使得 S 是 T 的前缀,多组数据。
【输入】
第一行一个整数 T,表示数据组数。
对于每组数据,第一行一个数 n,接下来 n 行输入 n 个数字串。(对于 100% 的数据,1≤T≤40,1≤n≤104 。)
【输出】
对于每组数据,若存在两个数字串 S,T,使得 S 是 T 的前缀,则输出 NO ,否则输出 YES 。
请注意此处结果与输出的对应关系!
【输入样例】
2
3
911
97625999
91125426
5
113
12340
123440
12345
98346
【输出样例】
NO
YES
用插入函数模板,原来的find函数是用来查找是否存在一模一样的,不合适了
题目要求是否存在一个字符串是另一个的前缀
我们直接一边插入一边判断,判断是否有重复的地方
两种情况,一种是原来的树上已经插入过的树枝长,直接用x记录数组不等于0的个数(x一定的等于新插入的字符串长度)
一种是原来树上的树枝短,需要在循环的时候判断,中间某个点是否已经等于1(exist数组)
写的时候忘了更新cnt,把所有树都合一起了,半天没找着错,老是运行错误,气死了😑😑
#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<string>using namespace std;int t,n;
string s;
int tr[100010][10];
int cnt=0;
int exist[100010];//储存该结点的结尾的字符串是否存在
int insert(string s)//插入函数,在插入的时候直接判断
{int p=0,x=0,y=0;int l=s.length();for(int i=0;i<l;i++){int c=s[i]-'0';if(tr[p][c]!=0) x++;//标记有多少重复 //当树上串长且完全重复,符合条件if(tr[p][c]==0) tr[p][c]=++cnt;//没有就申请新的结点p=tr[p][c];if(exist[p]==1) y=l;//如果已有,则说明树上串短}exist[p]=1;if(x==l||y==l) return 1;else return 0;
}int main()
{cin>>t;while(t--){cnt=0;//千万别忘了更新memset(tr,0,sizeof(tr));memset(exist,0,sizeof(exist));cin>>n;int flag=0;for(int i=1;i<=n;i++){cin>>s;if(insert(s)) flag=1;}if(flag==0) cout<<"YES"<<endl;else cout<<"NO"<<endl;}return 0;
}