后缀字符串|计蒜客2019蓝桥杯省赛 B 组模拟赛（一）

样例输入：
3
ba
a
aba
样例输出：
2
3
1

思路一：暴力，只能过50%数据，枚举每一个字符串，内层枚举其他字符串判断是否以这个字符串为后缀

思路二：哈希表，存储每一个后缀的数目，string.substr函数取后缀

substr用法:

代码一：

#include <bits/stdc++.h>
using namespace std;

string s[10010];
int n;

int main(){
	cin>>n;
	for(int i = 0;i<n;i++){
		cin>>s[i];
	}
	//ba a aba 
	//枚举每一个字符串 
	for(int i=0;i<n;i++){
		int ans = 0;
		int leni = s[i].size();
		//枚举其他字符串 
		for(int j=0;j<n;j++){
			if(i==j){
				ans++;
				continue;
			}
			int lenj = s[j].size();
			if(lenj < leni){
				continue;
			}
			//判断其他字符串是否以这个字符串为后缀 
			int flag = 1;
			int p = leni - 1;
			int q = lenj - 1;
			//从i字符串的末尾遍历到i结束 从j的字符串末尾遍历 到j结束如果不相等就跳出
			for(;p>=0,q>=0;p--,q--){
				if(p<0 || q<0)break; //这里什么bug 指针内存错误 
				if(s[i][p] != s[j][q]){
					flag = 0;
					break;
				}
			}
			if(flag) ans++;
		}
		cout<<ans<<endl;
	}
	return 0;
} 

代码二：

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int INF = 0x3f3f3f3f;
const LL mod = 1e9 + 7;
const int N = 100005;
string a[N];

/*使用map:统计数目*/ 
/*string.substr(i) 取*/ 

int main() {
    map<string, int> mp;
    int n;
    cin >> n;
    //输入每一个字符串 
    for (int i = 0; i < n; i++) {
        cin >> a[i];
        //对输入的当前字符串作 substr取每一个后缀; 并更新当前后缀在mp中的数量 
        for (int j = 0; j < a[i].size(); j++) {
            mp[a[i].substr(j)]++;
        }
    }
    for (int i = 0; i < n; i++) {
        cout << mp[a[i]] << endl;
    }
    return 0;
}