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 <iostream>
#include <algorithm>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <cstring>
#include <string>
#include <set>
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;
}