AtCoder Beginner Contest 287 D - Match or Not

原题：D - Match or Not

题目：给你字符串S和T（保证S的长度大于T的长度），问S'和T是否匹配。

    匹配的定义：用任意字符替换字符串X和Y中的'?'后，如果字符串相同则X与Y匹配。

对于x=0，1，2，…，|T|：
     S'由S的前x个字符与后（|T|-x）个字符连接而成。

题目数据范围1 <= |T| <= |S| <= 3e5，显然不能暴力。仔细观察发现其实只需要从头开始统计出S与T相同的字符数，再从后面统计出S和T相同的字符数。

int cnt1 = 0, cnt2 = 0;   //cnt1记录S的前面部分与T有多少个相同字符，cnt2记录后面部分
for (int i = 0; i < s2.size(); i++){
    if (s1[i] == s2[i] || s1[i] == '?' || s2[i] == '?') cnt1++;
    else break;
}
for (int i = s1.size() - 1, j = s2.size() - 1; j >= 0 && i >= 0; i--, j--){
    if (s1[i] == s2[j] || s1[i] == '?' || s2[j] == '?') cnt2++;
    else break;
}

如果x同时满足x <= cnt1 和 (|T| - x) <= cnt2，说明对于当前的x所截取出来的S'与T是匹配的。

AC代码：

#include <bits/stdc++.h>
using namespace std;
int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    string s1, s2;
    cin >> s1 >> s2;
    int cnt1 = 0, cnt2 = 0;
    for (int i = 0; i < s2.size(); i++)
    {
        if (s1[i] == s2[i] || s1[i] == '?' || s2[i] == '?')
            cnt1++;
        else
            break;
    }
    for (int i = s1.size() - 1, j = s2.size() - 1; j >= 0 && i >= 0; i--, j--)
    {
        if (s1[i] == s2[j] || s1[i] == '?' || s2[j] == '?')
            cnt2++;
        else
            break;
    }
    for (int i = 0; i <= s2.size(); i++)
    {
        if (i <= cnt1 && s2.size() - i <= cnt2)
            cout << "Yes\n";
        else
            cout << "No\n";
    }
    return 0;
}