結果

問題 No.2858 Make a Palindrome
ユーザー shinchanshinchan
提出日時 2024-08-25 14:52:07
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 5,306 bytes
コンパイル時間 2,158 ms
コンパイル使用メモリ 211,564 KB
実行使用メモリ 42,028 KB
最終ジャッジ日時 2024-08-25 14:52:13
合計ジャッジ時間 5,859 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 WA -
testcase_02 WA -
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 AC 36 ms
6,940 KB
testcase_16 AC 38 ms
6,944 KB
testcase_17 AC 37 ms
6,940 KB
testcase_18 AC 36 ms
6,944 KB
testcase_19 AC 34 ms
6,940 KB
testcase_20 AC 33 ms
37,064 KB
testcase_21 AC 33 ms
37,804 KB
testcase_22 AC 20 ms
21,416 KB
testcase_23 AC 31 ms
36,916 KB
testcase_24 AC 28 ms
32,972 KB
testcase_25 AC 25 ms
29,248 KB
testcase_26 AC 26 ms
27,968 KB
testcase_27 AC 36 ms
41,500 KB
testcase_28 AC 28 ms
31,092 KB
testcase_29 AC 33 ms
38,652 KB
testcase_30 AC 36 ms
42,028 KB
testcase_31 AC 36 ms
41,892 KB
testcase_32 AC 35 ms
42,000 KB
testcase_33 AC 36 ms
41,984 KB
testcase_34 AC 37 ms
41,972 KB
testcase_35 AC 37 ms
41,884 KB
testcase_36 AC 36 ms
41,928 KB
testcase_37 AC 37 ms
42,024 KB
testcase_38 AC 36 ms
41,924 KB
testcase_39 AC 36 ms
42,000 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#define all(v) (v).begin(),(v).end()
#define pb(a) push_back(a)
#define rep(i, n) for(int i=0;i<n;i++)
#define foa(e, v) for(auto& e : v)
using ll = long long;
const ll MOD7 = 1000000007, MOD998 = 998244353, INF = (1LL << 60);
#define dout(a) cout<<fixed<<setprecision(10)<<a<<endl;

struct RollingHash {
    static const int base1 = 1007, base2 = 2009;
    static const int mod1 = 1000000007, mod2 = 1000000009;
    std::vector<long long> hash1, hash2, power1, power2;
    std::vector<long long> hash1_, hash2_, power1_, power2_;

    void rev_init(std::string S) {
        std::reverse(S.begin(), S.end());
        int n = (int)S.size();
        hash1_.assign(n+1, 0);
        hash2_.assign(n+1, 0);
        power1_.assign(n+1, 1);
        power2_.assign(n+1, 1);
        for (int i = 0; i < n; ++i) {
            hash1_[i+1] = (hash1_[i] * base1 + S[i]) % mod1;
            hash2_[i+1] = (hash2_[i] * base2 + S[i]) % mod2;
            power1_[i+1] = (power1_[i] * base1) % mod1;
            power2_[i+1] = (power2_[i] * base2) % mod2;
        }
    }
 
    // construct
    RollingHash(const std::string &S, int rev = 0) {
        int n = (int)S.size();
        hash1.assign(n+1, 0);
        hash2.assign(n+1, 0);
        power1.assign(n+1, 1);
        power2.assign(n+1, 1);
        for (int i = 0; i < n; ++i) {
            hash1[i+1] = (hash1[i] * base1 + S[i]) % mod1;
            hash2[i+1] = (hash2[i] * base2 + S[i]) % mod2;
            power1[i+1] = (power1[i] * base1) % mod1;
            power2[i+1] = (power2[i] * base2) % mod2;
        }
        if(rev) rev_init(S);
    }
    
    // get hash of S[left:right)
    inline std::pair<long long, long long> get(int l, int r) const {
        long long res1 = hash1[r] - hash1[l] * power1[r-l] % mod1;
        if (res1 < 0) res1 += mod1;
        long long res2 = hash2[r] - hash2[l] * power2[r-l] % mod2;
        if (res2 < 0) res2 += mod2;
        return {res1, res2};
    }

    inline std::pair<long long, long long> get_rev(int l, int r) const {
        long long res1 = hash1_[r] - hash1_[l] * power1_[r-l] % mod1;
        if (res1 < 0) res1 += mod1;
        long long res2 = hash2_[r] - hash2_[l] * power2_[r-l] % mod2;
        if (res2 < 0) res2 += mod2;
        return {res1, res2};
    }

    // [l, r) is palindrome
    inline bool is_palindrome(int l, int r, int n) const {
        int l_rev = n - r;
        int r_rev = n - l;
        return get(l, r) == get_rev(l_rev, r_rev);
    }
 
    // get lcp of S[a:) and T[b:)
    inline int getLCP(int a, int b) const {
        int len = std::min((int)hash1.size()-a, (int)hash1.size()-b);
        int low = -1, high = len;
        while (high - low > 1) {
            int mid = (low + high) / 2;
            if (get(a, a+mid) != get(b, b+mid)) high = mid;
            else low = mid;
        }
        return low;
    }
};

void solve() {
    ll n, m;
    string s; 
    cin >> n >> m >> s;
    if(n >= m) {
        RollingHash r(s, 1);
        bool ok = 0;
        rep(i, n - m + 1) {
            if(r.get(i, i + m) == r.get_rev(i, i + m)) {
                ok = true;
                break;
            }
        }
        if(ok) {
            cout << 1 << endl;
            return;
        }
    } 
    ll ans = INF;
    if(n * 2 >= m) {
        RollingHash r(s + s, 1);
        bool ok = 0;
        rep(i, n * 2 - m + 1) {
            if(r.get(i, i + m) == r.get_rev(i, i + m)) {
                ok = true;
                break;
            }
        }
        if(ok) {
            cout << 1 << endl;
            return;
        }
    }
    s = s + s + s;
    int sz = s.size();
    RollingHash r(s, 1);
    
    {
        ll k = (m + 1) / 2;
        k *= 2;
        for(int i = n; i < (n << 1); i ++) {
            ll num = min(i, sz - i);
            if(r.is_palindrome(i - num, i + num, n * 3)) {
                ll val = 1;
                ll ri = sz - i - n;
                if(k / 2 > ri) {
                    ll r = k / 2 - ri;
                    val += (r + n - 1) / n;
                }
                ll le = i - n;
                if(k / 2 > le) {
                    ll l = k / 2 - le;
                    val += (l + n - 1) / n;
                }
                ans = min(ans, val);
            }
        }
    }
    {
        ll k = (m + 1) / 2;
        k *= 2;
        k ++;
        for(int i = n; i < (n << 1); i ++) {
            ll num = min(i, sz - i - 1);
            // cout << i << " " << num << endl;
            // if(r.get(i - num, i + num + 1) == r.get_rev(i - num, i + num + 1))
            if(r.is_palindrome(i - num, i + num + 1, n * 3)) {
                ll val = 1;
                ll ri = sz - i - 1 - n;
                if(k / 2 > ri) {
                    ll r = k / 2 - ri;
                    val += (r + n - 1) / n;
                }
                ll le = i - n;
                if(k / 2 > le) {
                    ll l = k / 2 - le;
                    val += (l + n - 1) / n;
                }
                // cout << i << " " << val << endl;
                ans = min(ans, val);
            }
        }
    }
    cout << (ans == INF ? -1 : ans) << endl;
        
}

int main() {
    cin.tie(0);
    ios::sync_with_stdio(false);
    int t; cin >> t;
    rep(i, t) solve();
    return 0;
}
0