#include<bits/stdc++.h>
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
using namespace std;
template<class A, class B>
ostream& operator<<(ostream& os, const pair<A, B> &p) {
os << '(' << p.first << ',' << p.second << ')';
return os;
}
template<class T>
ostream& operator<<(ostream& os, const vector<T> &v) {
os << '{';
bool osu = 1;
for(auto &i : v) { if(!osu) os << ','; os << i; osu = 0; }
os << '}';
return os;
}
template<class T>
ostream& operator<<(ostream& os, const deque<T> &v) {
os << '{';
bool osu = 1;
for(auto &i : v) { if(!osu) os << ','; os << i; osu = 0; }
os << '}';
return os;
}
template<class T,long long unsigned int size>
ostream& operator<<(ostream& os, const array<T,size> &arr) {
os << '{';
bool osu = 1;
for(auto &i : arr) { if(!osu) os << ','; os << i; osu = 0; }
os << '}';
return os;
}
#define ll long long
#define ull unsigned long long
#define lll __int128
#define rep(i,n,N) for(int i=(n);i<=(N);++i)
#define For(i,n,N) for(int i=(n);i< (N);++i)
#define rap(i,n,N) for(int i=(n);i>=(N);--i)
#define rip(i,n,N,V) for(int i=(n);i<=(N);i+=V)
#define mp make_pair
#define pb push_back
#define pob pop_back
#define pf push_front
#define pof pop_front
//#define le left
//#define ri right
#define fi first
#define se second
//#define ff fi.fi
//#define fs fi.se
//#define sf se.fi
//#define ss se.se
#define lc (id<<1)
#define rc ((id<<1)|1)
#define debug(x) cout << #x << " -> " << (x) << endl
#define out(x,y) cout << ">> " << (x) << " " << (y) << endl
#define ou3(x,y,z) cout << ">> " << (x) << " " << (y) << " " << (z) << endl
#define mems(x,y) memset(&x,y,sizeof x)
#define memsv(x,y) memset(&x[0], y, sizeof(x[0]) * x.size())
#define popcount __builtin_popcountll
#define clz(x) (1<<(31-__builtin_clz(x)))
#define clzll(x) (1LL<<(63-__builtin_clzll(x)))
#define all(x) x.begin(),x.end()
#define rsort(x) sort(x), reverse(x)
#define revsort(a,b) sort(a,b), reverse(a,b)
#define PQ(T) priority_queue<T>
#define PQr(T) priority_queue<T, vector<T>, greater<T>>
#define permute(x) while(next_permutation(all(x)))
#define distinct(x) x.erase(unique(all(x)),x.end())
#define vi vector<int>
#define vvi vector<vi>
#define vii vector<pii>
#define vll vector<ll>
#define vlll vector<pll>
#define vvll vector<vll>
#define ari(x) array<int,x>
#define arll(x) array<ll,x>
#define ard(x) array<double,x>
#define pii pair<int,int>
#define pll pair<ll,ll>
#define plll pair<ll,pll>
#define pllll pair<pll,pll>
#define piii pair<int,pii>
#define piiii pair<pii,pii>
#define psi pair<string,int>
#define endl '\n'
const int MAX = 3e3+5;
const int MOD = 2010408371;
const ll MOD2 = 2010405347;
const ll INF = 2e18;
const int dr[]={1,-1,0,0,1,1,-1,-1};
const int dc[]={0,0,-1,1,1,-1,1,-1};
const double pi = acos(-1);
const double EPS = 1e-9;
const int block = 2e3;
inline ll pw(ll x,ll y,ll md=MOD){ll ret=1;x%=md;while(y){if(y&1)ret=ret*x%md;x=x*x%md,y>>=1;}return ret;}
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
int n,m;
ll x[MAX],y[MAX],hs[MAX],key = 724729;
map<int,int> g;
string s,t;
ari(3) solve(){
rep(i,1,n)x[i] = (x[i-1]*key + s[i-1]) % MOD;
rep(i,1,m)y[i] = (y[i-1]*key + t[i-1]) % MOD;
int le,ri,mid,res,val;
bool st;
int a,b;
le = 1, ri = min(n,m);
while(le<=ri){
mid = le+ri>>1;
g.clear();
rep(i,mid,n)g[(x[i]+x[i-mid]*hs[mid])%MOD] = i;
st = 0;
rep(i,mid,m){
val = (y[i]+y[i-mid]*hs[mid])%MOD;
if(g.count(val)){
a = g[val]-mid, b = i-mid;
st = 1;
break;
}
}
if(st)res = mid, le = mid+1;
else ri = mid-1;
}
return {res, a, b};
}
int main(){
// cout<<fixed<<setprecision(10);
// freopen("input1.txt", "r", stdin);
// freopen("output.txt","w",stdout);
ios_base::sync_with_stdio(false); cin.tie(0); cout.tie(0);
cin>>s>>t;
n = s.size();
m = t.size();
int o = max(n,m);
hs[0] = MOD-1;
rep(i,1,o)hs[i] = hs[i-1]*key % MOD;
ari(3) ans = solve(), tmp;
reverse(all(t));
tmp = solve();
if(tmp[0]>ans[0]){
ans = tmp;
ans[2] = m-(ans[2]+ans[0]);
}
cout<<ans[0]<<endl;
cout<<ans[1]<<" "<<ans[2]<<endl;
return 0;
}
Compilation message
necklace.cpp: In function 'std::array<int, 3> solve()':
necklace.cpp:116:11: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
116 | mid = le+ri>>1;
| ~~^~~
necklace.cpp:131:19: warning: 'b' may be used uninitialized in this function [-Wmaybe-uninitialized]
131 | return {res, a, b};
| ^
necklace.cpp:113:6: warning: 'a' may be used uninitialized in this function [-Wmaybe-uninitialized]
113 | int a,b;
| ^
necklace.cpp:111:16: warning: 'res' may be used uninitialized in this function [-Wmaybe-uninitialized]
111 | int le,ri,mid,res,val;
| ^~~
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Partially correct |
2 ms |
348 KB |
Output is partially correct |
| 2 |
Partially correct |
2 ms |
348 KB |
Output is partially correct |
| 3 |
Partially correct |
10 ms |
664 KB |
Output is partially correct |
| 4 |
Partially correct |
9 ms |
604 KB |
Output is partially correct |
| 5 |
Partially correct |
10 ms |
668 KB |
Output is partially correct |
| 6 |
Partially correct |
11 ms |
616 KB |
Output is partially correct |
| 7 |
Partially correct |
7 ms |
600 KB |
Output is partially correct |
| 8 |
Partially correct |
7 ms |
604 KB |
Output is partially correct |
| 9 |
Partially correct |
9 ms |
652 KB |
Output is partially correct |
