|
|
вернуться в форумWA on test case 5 I need help, I got WA on test case 5, and I do not know what's the problem. Here my code: #include <bits/stdc++.h> using namespace std; string getPrefix(string s, int n){ string result; if(n >= s.size())result = s; else result = s.substr(0,n); return result; } string getSuffix(string s, int n){ int newN; string result; if(n >= s.size()) result = s; else result = s.substr(s.size() - n); return result; } int main(){ string s, a, b; int m; while(cin>>s>>a>>b>>m){ vector<pair<string, string> > v(m + 10, {"",""}); vector<long long int> cnt(m + 10, 0); v[1] = {a, a}; cnt[1] = 0; v[2] = {b, b}; cnt[2] = 0; long long int maxi = 0; for(int i = 3, from, to; i <= m + 2;i++){ cin>>from>>to; string text = v[from].second + v[to].first; string text2 = v[from].first + v[to].second; string prefix = getPrefix(text, s.size() - 1); string suffix = getSuffix(text, s.size() - 1); string prefixNew = getPrefix(text2, s.size() - 1); string suffixNew = getSuffix(text2, s.size() - 1); v[i] = {prefixNew, suffixNew}; cnt[i] = (cnt[from] + cnt[to]) % 1000000007; if(text.size() >= s.size()){ for(int j = 0 ; j < text.size() - s.size() + 1; j++){ if(s == text.substr(j,s.size()) ) cnt[i] = (cnt[i] + 1) % 1000000007; } } } long long int sol = 0; for(int i = 1; i <= m + 2; i++){ sol = max(sol, cnt[i]); } cout<<sol<<endl; } return 0; } |
|
|