SDNU-1099.前缀判断(set)

SDNU-1099.前缀判断

Description
给定 n 个字符串,求有多少字符串是其他字符串的前缀。
Input
第一行为一个整数n(1 <= n <= 1000),之后n行,每行一个字符串,字符串只由26个小写字母组成,最大长度为100。
Output
一个整数,有多少字符串是其他字符串的前缀。
Sample Input
5
abcde
ab
bcde
b
cde
Sample Output
2
最关键用到了set中的count和find的区别:
count返回的是次数,在set中就是0,1;
find返回的是迭代器。

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
using namespace std;
typedef long long ll;
#define INF 0x3f3f3f3f
int main()
{
    string s[1005];
    set<string> chu;//储存重复的
    set<string> p;//判断是否有重复的字符串
    set<string> q;//储存所有s[i]的所有前缀
    int n;
    cin>>n;
    int sum=0;
    for (int i=0;i<n;i++)
    {
        cin>>s[i];
        if (p.count(s[i]))
            chu.insert(s[i]);
        p.insert(s[i]);
        for (int j=1;j<s[i].length();j++)
        {
            string str=s[i].substr(0,j);//把s[i]所有可能的前缀都列出来,然后存起来
            q.insert(str);
        }
    }
    for (int i=0;i<n;i++)
        if (q.count(s[i])||chu.count(s[i]))
            sum++;
    cout<<sum<<endl;
    return 0;
}

你可能感兴趣的:(SDNU)