結果
問題 | No.2231 Surprising Flash! |
ユーザー |
![]() |
提出日時 | 2023-02-26 19:49:08 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 1,125 ms / 4,000 ms |
コード長 | 10,312 bytes |
コンパイル時間 | 4,943 ms |
コンパイル使用メモリ | 263,992 KB |
最終ジャッジ日時 | 2025-02-10 23:40:46 |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 44 |
コンパイルメッセージ
main.cpp: In function ‘std::string solve()’: main.cpp:346:16: warning: ‘idx’ may be used uninitialized [-Wmaybe-uninitialized] 346 | int lst=-INF,idx; | ^~~
ソースコード
#pragma region template#pragma GCC optimize("Ofast")#include <bits/stdc++.h>using namespace std;using ll=long long;using ld=long double;using vi=vector<int>;using vll=vector<ll>;using pi=pair<int,int>;using pll=pair<ll,ll>;#define overload2(a,b,c,...) c#define overload3(a,b,c,d,...) d#define overload4(a,b,c,d,e,...) e#define overload5(a,b,c,d,e,f,...) f#define TYPE1(T) template<typename T>#define TYPE2(T,U) template<typename T,typename U>#define TYPE(...) overload2(__VA_ARGS__,TYPE2,TYPE1)(__VA_ARGS__)#define TYPES1(T) template<typename... T>#define TYPES2(H,T) template<typename H,typename... T>#define TYPES(...) overload2(__VA_ARGS__,TYPES2,TYPES1)(__VA_ARGS__)#define REP4(i,s,n,d) for(int i=(s);i<(n);i+=(d))#define REP3(i,s,n) REP4(i,s,n,1)#define REP2(i,n) REP3(i,0,n)#define REP1(n) REP2(tomato,n)#define REP(...) overload4(__VA_ARGS__,REP4,REP3,REP2,REP1)(__VA_ARGS__)#define RREP4(i,n,s,d) for(int i=(n)-1;i>=(s);i-=(d))#define RREP3(i,n,s) RREP4(i,n,s,1)#define RREP2(i,n) RREP3(i,n,0)#define RREP1(n) RREP2(tomato,n)#define RREP(...) overload4(__VA_ARGS__,RREP4,RREP3,RREP2,RREP1)(__VA_ARGS__)#define FOR4(a,b,c,d,v) for(auto [a,b,c,d]:v)#define FOR3(a,b,c,v) for(auto [a,b,c]:v)#define FOR2(a,b,v) for(auto [a,b]:v)#define FOR1(a,v) for(auto a:v)#define FOR(...) overload5(__VA_ARGS__,FOR4,FOR3,FOR2,FOR1)(__VA_ARGS__)#define AFOR4(a,b,c,d,v) for(auto&[a,b,c,d]:v)#define AFOR3(a,b,c,v) for(auto&[a,b,c]:v)#define AFOR2(a,b,v) for(auto&[a,b]:v)#define AFOR1(a,v) for(auto&a:v)#define AFOR(...) overload5(__VA_ARGS__,AFOR4,AFOR3,AFOR2,AFOR1)(__VA_ARGS__)#define CFOR4(a,b,c,d,v) for(const auto&[a,b,c,d]:v)#define CFOR3(a,b,c,v) for(const auto&[a,b,c]:v)#define CFOR2(a,b,v) for(const auto&[a,b]:v)#define CFOR1(a,v) for(const auto&a:v)#define CFOR(...) overload5(__VA_ARGS__,CFOR4,CFOR3,CFOR2,CFOR1)(__VA_ARGS__)#define PFOR(v) for(bool f=true; (f ? exchange(f,false) : next_permutation(v));)#define ALL(v) v.begin(),v.end()#define RALL(v) v.rbegin(),v.rend()#define SORT(v) sort(ALL(v))#define RSORT(v) sort(RALL(v))#define REVERSE(v) reverse(ALL(v))#define UNIQUE(v) SORT(v),v.erase(unique(ALL(v)),v.end())#define SZ(v) int(v.size())TYPES(T) void input(T&... a){ (cin>>...>>a); }#define DECLARE(T,...) T __VA_ARGS__;input(__VA_ARGS__);#define INT(...) DECLARE(int,__VA_ARGS__)#define STR(...) DECLARE(string,__VA_ARGS__)#define LL(...) DECLARE(long long,__VA_ARGS__)#define CHR(...) DECLARE(char,__VA_ARGS__)#define DBL(...) DECLARE(double,__VA_ARGS__)#define VI(n,v) vi v(n);cin>>v;#define VLL(n,v) vll v(n);cin>>v;TYPE(T) istream&operator>>(istream&is,vector<T>&v){for(auto&a:v)cin>>a;return is;}TYPE(T) ostream&operator<<(ostream&os,const vector<T>&v){if(&os==&cerr)os<<"[";REP(i,v.size()){os<<v[i];if(i+1<v.size())os<<(&os==&cerr?",":" ");}if(&os==&cerr)os<<"]";return os;}TYPE(T,S) istream&operator>>(istream&is,pair<T,S>&p){cin>>p.first>>p.second;return is;}#ifdef __DEBUG#include <debug>#else#define debug(...) void(0)#endifvoid print(){ cout << '\n'; }TYPES(T,Ts) void print(const T& a,const Ts&... b){cout<<a;(cout<<...<<(cout<< ' ',b));cout << '\n';}TYPE(T) using pq=priority_queue<T>;TYPE(T) using pqg=priority_queue<T,vector<T>,greater<T>>;TYPE(T) T pick(queue<T>& que){assert(que.size()); T a=que.front();que.pop();return a;}TYPE(T) T pick(pq<T>& que){assert(que.size()); T a=que.top();que.pop();return a;}TYPE(T) T pick(pqg<T>& que){assert(que.size()); T a=que.top();que.pop();return a;}TYPE(T) T pick(stack<T>& sta){assert(sta.size()); T a=sta.top();sta.pop();return a;}TYPE(T) void clear(T&v){ v=decltype(v)(); }string YES(bool f=true){return (f?"YES":"NO");}string Yes(bool f=true){return (f?"Yes":"No");}string yes(bool f=true){return (f?"yes":"no");}constexpr int INF=1e9+7;constexpr ll LINF=ll(1e18)+7;constexpr ld EPS=1e-10;vi iota(int n){vi a(n);iota(ALL(a),0);return a;}TYPE(T) vector<pair<T,int>> query_sort(const vector<T>&v){vector<pair<T,int>> res(v.size());REP(i,v.size())res[i]={v[i],i};SORT(res);return res;}TYPE(T) T rev(T a){ REVERSE(a);return a; }TYPE(T) void fin(T a){cout<<a<<'\n';exit(0);}TYPE(T) bool chmax(T &a,T b){return (a<b&&(a=b,true));}TYPE(T) bool chmin(T &a,T b){return (a>b&&(a=b,true));}TYPES(T,Ns) auto make_vector(T x,int n,Ns ...ns){if constexpr(sizeof...(ns)==0)return vector<T>(n,x);else return vector(n,make_vector<T>(x,ns...));}bool in(const ll S,const int a){return (S>>a)&1;}int popcount(const ll S){return __builtin_popcountll(S);}int digit(char c){ return (c>='0' and c<='9' ? c-'0' : -1);}ll sqrtll(ll a){for(ll b=sqrt(a);b*b<=a;b++)if(b*b==a)return b;for(ll b=sqrt(a);b>=0 and b*b>=a;b--)if(b*b==a)return b;return -1;}#pragma endregion template#include <atcoder/convolution>using namespace atcoder;struct Modint61{using ll=long long;static constexpr ll MOD=(1LL<<61)-1;ll v;Modint61(const ll x=0):v(x){if(v<0||v>=MOD){v%=MOD;if(v<0)v+=MOD;}}static Modint61 raw(ll v){Modint61 x;x.v=v;return x;}private:template<int d>static constexpr pair<ll,ll> divide(const ll&a){return {a>>d, a&((1LL<<d)-1)};}public:Modint61 pow(long long k){Modint61 res(1),tmp(v);while(k){if(k&1)res*=tmp;tmp*=tmp;k>>=1;}return res;}Modint61 inv(){ return pow(MOD-2); }Modint61& operator+=(Modint61 a){v+=a.v;if(v>=MOD)v-=MOD;return *this;}Modint61& operator-=(Modint61 a){v+=MOD-a.v;if(v>=MOD)v-=MOD;return *this;}Modint61& operator*=(Modint61 a){const auto [vu,vd]=divide<31>(v);const auto [au,ad]=divide<31>(a.v);ll x = vd*au + vu*ad;const auto [xu,xd]=divide<30>(x);v = vu*au*2 + xu + (xd<<31) + vd*ad;if(v>=MOD){const auto [vu2,vd2]=divide<61>(v);v=vu2+vd2;}if(v>=MOD)v-=MOD;return *this;}Modint61& operator/=(Modint61 a){return (*this)*=a.inv();}Modint61 operator+(Modint61 a) const{return Modint61(v)+=a;}Modint61 operator-(Modint61 a) const{return Modint61(v)-=a;}Modint61 operator*(Modint61 a) const{return Modint61(v)*=a;}Modint61 operator/(Modint61 a) const{return Modint61(v)/=a;}Modint61 operator+() const{return *this;}Modint61 operator-() const{return v?Modint61(MOD-v):Modint61(v);}bool operator==(const Modint61 a)const{return v==a.v;}bool operator!=(const Modint61 a)const{return v!=a.v;}bool operator<(const Modint61 a)const{return v<a.v;}};template<char MARGIN>struct ForString{static constexpr char change(char c){ return c-MARGIN; }static constexpr char restore(char a){ return a+MARGIN; }static vector<char> change(const string&s){vector<char> v(s.size());for(int i=0;i<s.size();i++)v[i]=change(s[i]);return v;}static string restore(const vector<char>&v){string s(v.size(),'#');for(int i=0;i<v.size();i++)s[i]=restore(v[i]);return s;}};struct FSAa{static constexpr char change(char c){ return c<='Z'?c-'A':26+c-'a'; }static constexpr char restore(char a){ return a<26?'A':a-26+'a'; }static vector<char> change(const string&s){vector<char> v(s.size());for(int i=0;i<s.size();i++)v[i]=change(s[i]);return v;}static string restore(const vector<char>&v){string s(v.size(),'#');for(int i=0;i<v.size();i++)s[i]=restore(v[i]);return s;}};using FSA=ForString<'A'>;using FSa=ForString<'a'>;using FS0=ForString<'0'>;#ifdef STR#define STRA(s) STR(s##tomato);auto s=FSA::change(s##tomato);#define STRa(s) STR(s##tomato);auto s=FSa::change(s##tomato);#define STR0(s) STR(s##tomato);auto s=FS0::change(s##tomato);#endifrandom_device rollonghash_rnd;mt19937 rollonghash_mt(rollonghash_rnd());template<typename CHAR=char,typename MINT=Modint61>class RollingHash{using ll=long long;static constexpr ll mod = (1LL<<61)-1;using T=ll;inline static const MINT base=MINT::raw(rollonghash_mt()%(mod-2)+2);int n;vector<MINT> hash,power;public:RollingHash(const vector<CHAR>&v):n(v.size()),hash(n+1,0),power(n+1,1){for(int i=0;i<n;i++){hash[i+1]=nxt_hash(hash[i],v[i]);power[i+1]=power[i]*base;}}static MINT nxt_hash(MINT x,CHAR c){ return (x*base) + MINT::raw(c); }T get_hash(int l=0,int r=-1){if(r<0)r=n;return (hash[r]-hash[l]*power[r-l]).v;}static T full_hash(const vector<CHAR>&v){MINT res=0;for(const char&c:v)res=nxt_hash(res,c);return res.v;}};template<class F>long long binary_search_ll(long long ok,long long ng,const F &is_ok){while(abs(ok-ng)>1){long long mid=(ok+ng)>>1;(is_ok(mid)?ok:ng)=mid;}return ok;}vector<int> wild_pattern_matching(const string&T,const string&P){int n=T.size(),m=P.size();vector<ll> f(n),g(m);REP(i,n)f[i] = T[i]=='?' ? 0 : T[i]-'a'+1;REP(j,m)g[j] = P[j]=='?' ? 0 : P[j]-'a'+1;REVERSE(g);vector<ll> sum(n-(m-1),0);{auto F=f;AFOR(a,F)a=a*a*a;auto fg=convolution_ll(F,g);REP(i,n-(m-1))sum[i] += fg[i+m-1];}{auto F=f;AFOR(a,F)a=a*a;auto G=g;AFOR(a,G)a=a*a;auto fg=convolution_ll(F,G);REP(i,n-(m-1))sum[i] -= 2*fg[i+m-1];}{auto G=g;AFOR(a,G)a=a*a*a;auto fg=convolution_ll(f,G);REP(i,n-(m-1))sum[i] += fg[i+m-1];}debug(sum);vector<int> res;REP(i,n-(m-1))if(!sum[i])res.push_back(i);debug(res);return res;}string solve(){INT(n,m);STR(T,P);auto v=wild_pattern_matching(T,P);if(!v.size())return "-1";// 最初に [? が 非 a になる場所] が一番後のやつvector<char> tt(n),pp(m);REP(i,n)tt[i]=(T[i]=='?' ? 0 : FSa::change(T[i]));pp = FSa::change(P);RollingHash t(tt),p(pp);int lst=-INF,idx;CFOR(s,v){if(t.get_hash(s,s+m)==p.get_hash()){AFOR(c,T)if(c=='?')c='a';return T;}auto is_ok=[&](int x){return t.get_hash(s,s+x)==p.get_hash(0,x);};int x=binary_search_ll(0,m,is_ok);if(chmax(lst,s+x))idx=s;}REP(j,m)T[idx+j]=P[j];AFOR(c,T)if(c=='?')c='a';return T;}int main(){ios::sync_with_stdio(false);cin.tie(nullptr);int testsize;cin>>testsize;while(testsize--)cout<<solve()<<'\n';}