百度笔试题 字符串匹配

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

int main()
{
	string a, b;
	set<string> s;
	int cnt = 0;
	cin >> a >> b;
	for (int i=0; i<a.size(); i++) {
		if (i+b.size()-1 >= a.size())   //当分割出的字符串长度小于b字符串的长度
			continue;
		string cur = a.substr(i, b.size());  //分割
		if (s.count(cur))                 //已存在该种类，不继续下面的操作
			continue;
		s.insert(cur);                   //加入
		bool flag = true;
		for (int k=0; k<b.size(); k++) {   //开始匹配
			if (cur[k] != b[k]) {
				if (b[k] == '?') {
					continue;
				} else {
					flag = false;
					break;
				}
			}
		}
		if (flag)
			cnt++;
	}
	printf ("%d", cnt);
    return 0;
}