Submission #394496

#TimeUsernameProblemLanguageResultExecution timeMemory
394496ak2006Three Friends (BOI14_friends)C++14
0 / 100
970 ms51212 KiB
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using vb = vector<bool>;
using vvb = vector<vb>;
using vi = vector<int>;
using vvi = vector<vi>;
using vl = vector<ll>;
using vvl = vector<vl>;
using vc = vector<char>;
using vvc = vector<vc>;
const ll mod = 1e9 + 7,inf = 1e18,p = 31;
#define pb push_back
#define fast ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0);
int n = 2e6 + 5;
vi a(n),invp(n);
vl hsh(n,1),power(n,1);
ll poww(ll a,ll b)
{
    if (b == 0)return 1;
    if (b == 1)return a;
    ll ret = poww(a,b/2);
    if (b%2 == 0)return (ret * ret)%mod;
    return ((ret * ret)%mod * a)%mod;
}
ll inv(ll a)
{
    return (poww(a,mod - 2)%mod + mod)%mod;
}
ll get_hash(int a,int b)
{
    return (((hsh[b] - ((a >= 1 ? hsh[a - 1] : 0))%mod))%mod + mod)%mod;
}
int main()
{
    for (int i = 1;i<=n;i++)power[i] = (power[i - 1] * p)%mod,invp[i] = inv(power[i]);
    cin>>n;

    if (n % 2 == 0){cout<<"NOT POSSIBLE";return 0;}

    string s;
    cin>>s;

    for (int i = 0;i<n;i++)
        a[i] = s[i] - 'A' + 1;

    hsh[0] = a[0];
    ll val = -1;

    for (int i = 0;i<n - 1;i++)
        hsh[i + 1] = (poww(p,i + 1) * a[i + 1] + hsh[i])%mod;

    int start = -1,len = (n - 1)/2;

    for (int i = 0;i<n;i++){
        if (i == 0){

            ll v1 = ((hsh[len] - hsh[0])%mod + mod)%mod;

            ll v2 = ((hsh[n - 1] - hsh[len])%mod + mod)%mod;

            ll act = (v1 * invp[1])%mod;

            if ((v1 * power[len])%mod == v2){
                if (val != - 1 && act != val){cout<<"NOT UNIQUE";return 0;}
                if (val == -1)val = act;
                start = 1;
            }
        }
        else if (i < len){
            ll v1 = ((((hsh[i - 1] * p)%mod + (hsh[len] - hsh[i]))%mod)%mod + mod)%mod;
            ll v2 = ((hsh[n - 1] - hsh[len])%mod + mod)%mod;
            ll act = (v1 * invp[1])%mod;
            if ((v1 * power[len])%mod == v2){
                if (val != -1 && act != val){cout<<"NOT UNIQUE";return 0;}
                if (val == -1)val = act;
                start = len + 1;
            }
        }
        else{
            ll v1 = hsh[len - 1];
            ll v2 = ((((hsh[i - 1] - hsh[len - 1]) * p)%mod + (hsh[n - 1] - hsh[i])%mod)%mod + mod)%mod;
            ll act = v1;
            if ((v1 * power[len + 1])%mod == v2){
                if (val != -1 && val != act){cout<<"NOT UNIQUE";return 0;}
                if (val == -1)val = act;
                start = 0;
            }
        }
    }

    if (val == -1)cout<<"NOT POSSIBLE";
    else cout<<s.substr(start,len);
    return 0;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...