結果

問題 No.958 たぷりすたべる (回文)
ユーザー mtsd
提出日時 2019-12-21 00:26:32
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 366 ms / 2,000 ms
コード長 1,968 bytes
コンパイル時間 1,087 ms
コンパイル使用メモリ 116,100 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-07-18 05:09:13
合計ジャッジ時間 9,185 ms
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 53
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <algorithm>
#include <bitset>
#include <cassert>
#include <chrono>
#include <climits>
#include <cmath>
#include <complex>
#include <cstring>
#include <deque>
#include <functional>
#include <iostream>
#include <list>
#include <map>
#include <numeric>
#include <queue>
#include <random>
#include <set>
#include <stack>
#include <unordered_map>
#include <unordered_set>
#include <vector>
#include <cstdint>
using namespace std;
typedef long long ll;
#define MP make_pair
#define PB push_back
#define inf 1000000007
#define mod 1000000007
#define rep(i,n) for(int i = 0; i < (int)(n); ++i)

void manacher(const string& S,vector<int>& res)
{
    int sz = (int)S.size(), i = 0, j = 0, k;
    res.resize(sz);
    while(i < sz){
        while(i-j >= 0 && i+j < sz && S[i-j] == S[i+j]) j++;
        res[i] = j, k = 1;
        while(i-k >= 0 && i+k < sz && k+res[i-k] < j){
            res[i+k] = res[i-k], k++;
        }
        i += k; j -= k;
    }
}

int main(){
    ll n,k,q;
    cin >> n >> k >> q;
    string s;
    cin >> s;
    if(k==1){
        vector<int> p;
        manacher(s,p);
        rep(i,q){
            ll a;
            cin >> a;
            a--;
            cout << 2*p[a]-1 << "\n";
        }
    }else{
        s = s+s+s;
        ll len = n*k;
        vector<int> p;
        manacher(s,p);    
        int m = s.size();
        rep(i,q){
            ll a;
            cin >> a;
            a--;
            if(a<n){
                cout << 2*p[a]-1 << "\n";
            }else if(a >= len-n){
                a -= n*(k-1);
                cout << 2*p[2*n+a]-1 << "\n";
            }else{
                ll b = a%n;
                b+=n;
                ll L = b+1;
                ll R = 3*n-b;
                ll t = p[b];
                if(t==L||t==R){
                    cout << min(2*a+1,2*(len-a)-1) << "\n";
                }else{
                    cout << 2*t-1 << "\n";
                }
            }
        }
    }
    return 0;
}
0