結果
問題 | No.2626 Similar But Different Name |
ユーザー |
![]() |
提出日時 | 2024-03-10 00:21:20 |
言語 | C++17(gcc12) (gcc 12.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 345 ms / 3,000 ms |
コード長 | 1,924 bytes |
コンパイル時間 | 3,019 ms |
コンパイル使用メモリ | 236,748 KB |
実行使用メモリ | 68,708 KB |
最終ジャッジ日時 | 2024-09-29 21:24:01 |
合計ジャッジ時間 | 10,552 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 35 |
ソースコード
#include <bits/stdc++.h>#include <atcoder/convolution>using namespace std;using ll = long long;template <typename T>vector<int> z_algorithm(T &S){vector<int> Z(S.size());Z[0] = S.size();int i=1, j=0;while(i < S.size()){while(i+j < S.size() && S[j] == S[i+j]) j++;Z[i] = j;if (j == 0){i++;continue;}int k=1;while(k < j && k + Z[k] < j){Z[i+k] = Z[k];k++;}i += k;j -= k;}return Z;}int main(){cin.tie(nullptr);ios_base::sync_with_stdio(false);//SとTが大文字、小文字の区別なしで一致するかどうか?→Z-algorithm//S_iとT_iの大小が異なる箇所を計算する//h(c) = (大文字かどうか)としておき、//i=0, ..., N-Mに対してsum(j=0 to M-1)(h(S_i+j)-h(T_j))^2が異なる箇所の数//sum(j=0 to M-1) S_{i+j}T_jはU_{M-1-j}=T_jとして畳み込みすれば、M+i-1の係数ll N, M, K, cnt, ans=0;cin >> N >> M >> K;string S, T, S2="", T2="";cin >> S >> T;for (int i=0; i<N; i++){S2.push_back(tolower(S[i]));}for (int i=0; i<M; i++){T2.push_back(tolower(T[i]));}T2 += S2;vector<int> z = z_algorithm(T2);vector<ll> s(N), t(M), u(M), ss(N+1), tt(M+1);for (int i=0; i<N; i++){s[i] = ('A' <= S[i] && S[i] <= 'Z' ? 1 : 0);ss[i+1] = ss[i] + s[i];}for (int i=0; i<M; i++){t[i] = ('A' <= T[i] && T[i] <= 'Z' ? 1 : 0);tt[i+1] = tt[i] + t[i];}for (int i=0; i<M; i++){u[i] = t[M-i-1];}u = atcoder::convolution_ll(s, u);for (int i=0; i<N; i++){if (z[i+M] < M) continue;cnt = tt[M]+ss[i+M]-ss[i];cnt -= u[i+M-1]*2;ans += (1 <= cnt && cnt <= K);}cout << ans << endl;return 0;}