1414. Astronomical Database(STL)

1414

破题 又逼着用stl 卡内存 trie树太耗了 水不过去

用set存字符串 set可以自己按一定顺序存 且没有重复的 再用lower_bound二分查找字符串的第一次出现 接着往后找就行了

 1 #include <iostream>

 2 #include<cstdio>

 3 #include<cstring>

 4 #include<stdlib.h>

 5 #include<algorithm>

 6 #include<set>

 7 #include<string>

 8 using namespace std;

 9 #define N 10010

10 set<string>q;

11 set<string>::iterator it;

12 int kk;

13 string ys = "sun";

14 char ss[22];

15 bool cmp(string s)

16 {

17     int i,tk = (int)s.size();

18     if(tk<kk)

19     return false;

20     for(i = 0 ; i < kk ; i++)

21     {

22         if(s[i]!=ss[i])

23         return false;

24     }

25     return true;

26 }

27 int main()

28 {

29     char c;

30     q.insert(ys);

31     while(cin>>c)

32     {

33         cin>>ss;

34         if(c=='?')

35         {

36             cout<<ss<<endl;

37             kk = strlen(ss);

38             if(q.size())

39             {

40                 it = lower_bound(q.begin(),q.end(),ss);

41                 int k;

42                 for(k=0; it!=q.end()&&cmp(*it)&&k<20;it++,k++)

43                 cout<<"  "<<(*it)<<endl;

44             }

45         }

46         else

47         {

48             q.insert(ss);

49         }

50         getchar();

51     }

52     return 0;

53 }

View Code