#1005279 (C++23) No.2858 Make a Palindrome

提出ソース

結果

問題	No.2858 Make a Palindrome
ユーザー	kwm_t
提出日時	2024-08-26 19:34:58
言語	C++23 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	825 ms / 3,000 ms
コード長	4,109 bytes
コンパイル時間	5,841 ms
コンパイル使用メモリ	317,492 KB
実行使用メモリ	12,044 KB
最終ジャッジ日時	2024-08-26 19:35:30
合計ジャッジ時間	31,253 ms
ジャッジサーバーID （参考情報）	judge1 / judge5

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	825 ms 6,812 KB
testcase_01	AC	396 ms 6,944 KB
testcase_02	AC	396 ms 6,944 KB
testcase_03	AC	397 ms 6,944 KB
testcase_04	AC	398 ms 6,940 KB
testcase_05	AC	374 ms 6,940 KB
testcase_06	AC	330 ms 6,944 KB
testcase_07	AC	330 ms 6,944 KB
testcase_08	AC	329 ms 6,940 KB
testcase_09	AC	330 ms 6,940 KB
testcase_10	AC	594 ms 6,940 KB
testcase_11	AC	589 ms 6,940 KB
testcase_12	AC	593 ms 6,944 KB
testcase_13	AC	590 ms 6,940 KB
testcase_14	AC	593 ms 6,944 KB
testcase_15	AC	618 ms 6,940 KB
testcase_16	AC	624 ms 6,940 KB
testcase_17	AC	628 ms 6,940 KB
testcase_18	AC	627 ms 6,944 KB
testcase_19	AC	622 ms 6,944 KB
testcase_20	AC	592 ms 11,516 KB
testcase_21	AC	606 ms 11,540 KB
testcase_22	AC	310 ms 7,716 KB
testcase_23	AC	622 ms 11,644 KB
testcase_24	AC	523 ms 11,268 KB
testcase_25	AC	461 ms 11,020 KB
testcase_26	AC	436 ms 8,988 KB
testcase_27	AC	669 ms 11,912 KB
testcase_28	AC	500 ms 11,084 KB
testcase_29	AC	614 ms 11,564 KB
testcase_30	AC	694 ms 11,924 KB
testcase_31	AC	665 ms 11,924 KB
testcase_32	AC	643 ms 11,916 KB
testcase_33	AC	673 ms 11,920 KB
testcase_34	AC	671 ms 11,916 KB
testcase_35	AC	697 ms 11,916 KB
testcase_36	AC	669 ms 12,044 KB
testcase_37	AC	693 ms 11,924 KB
testcase_38	AC	665 ms 11,792 KB
testcase_39	AC	703 ms 11,920 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <bits/stdc++.h>
#include <atcoder/all>
//using namespace std;
//using namespace atcoder;
//using mint = modint1000000007;
//const int mod = 1000000007;
using mint = atcoder::modint998244353;
const int mod = 998244353;
//const int INF = 1e9;
const long long LINF = 1e18;
#define rep(i, n) for (int i = 0; i < (n); ++i)
#define rep2(i,l,r)for(int i=(l);i<(r);++i)
#define rrep(i, n) for (int i = (n) - 1; i >= 0; --i)
#define rrep2(i,l,r)for(int i=(r) - 1;i>=(l);--i)
#define all(x) (x).begin(),(x).end()
#define allR(x) (x).rbegin(),(x).rend()
#define P pair<int,int>
template<typename A, typename B> inline bool chmax(A & a, const B & b) { if (a < b) { a = b; return true; } return false; }
template<typename A, typename B> inline bool chmin(A & a, const B & b) { if (a > b) { a = b; return true; } return false; }
namespace RangeSumMint {
	// 区間加算
	mint op(mint a, mint b) { return a + b; }
	mint e() { return 0; }
	using segtree = atcoder::segtree<mint, op, e>;
}
template<typename T>
class Rollinghash {
public:
	Rollinghash(const std::vector<T>&v, int base) :base(base) {
		build(v);
	}
	Rollinghash(const std::string &s, int base = 27) :base(base) {
		std::vector<int>v;
		for (auto c : s)v.push_back(c - 'a' + 1);
		build(v);
	}
	// rは区間内に収まらなくてもいい
	// lは区間内::todo...
	mint get(const long long l, const long long r) {
		long long width = r - l;
		long long h = std::min(width, (long long)size - l);
		long long b = ((width - h) / size) * size;
		long long t = width - h - b;
		mint hash = 0;
		if (h) {
			//std::cout << l << "_" << l + h << std::endl;
			hash += seg.prod(l, l + h) / pow[l];
			//std::cout << hash.val() << std::endl;
		}
		if (b) {
			//std::cout << 0 << "!" << size << std::endl;
			hash += seg.prod(0, size) * (atcoder::pow_mod(base.val(), b, mod) - 1) / (pow[size] - 1) * pow[h];
			//std::cout << hash.val() << std::endl;
		}
		if (t) {
			//std::cout << 0 << "?" << t << std::endl;
			hash += atcoder::pow_mod(base.val(), h + b, mod)*seg.prod(0, t);
			//std::cout << hash.val() << std::endl;
		}
		return hash;
	}
	void update(int index, T value) {
		seg.set(index, value * pow[index]);
	}
	void update(int index, char c) {
		update(index, c - 'a' + 1);
	}
private:
	int size;
	mint base;
	mint inv;
	RangeSumMint::segtree seg;
	std::vector<mint>pow;

	template<typename U>
	void build(const std::vector<U>&v) {
		inv = base.inv();
		size = v.size();

		pow.resize(size + 1);
		pow[0] = 1;
		for (int i = 0; i < size; ++i)pow[i + 1] = pow[i] * base;

		std::vector<mint>mv(size);
		for (int i = 0; i < size; ++i) {
			mv[i] = pow[i] * v[i];
		}
		seg = RangeSumMint::segtree(mv);
	}
};
int main() {
	std::ios::sync_with_stdio(false);
	std::cin.tie(nullptr);
	int t; std::cin >> t;
	const int base = 100;
	std::vector<mint>pow(200005);
	pow[0] = 1;
	rep2(i, 1, pow.size())pow[i] = pow[i - 1] * base;
	while (t--) {
		int n, m; std::cin >> n >> m;
		std::string s; std::cin >> s;
		auto t = s;
		reverse(all(t));
		long long ans = LINF;
		auto rhS = Rollinghash<std::string>(s, 100);
		auto rhT = Rollinghash<std::string>(t, 100);
		rep(i, n) {
			int sz = (m) / 2;
			int iR = (i + 1) % n;
			mint hR = rhS.get(iR, iR + sz);
			int iL = (i - 1 + n) % n;
			iL = n - 1 - iL;
			//std::cout << std::endl;
			mint hL = rhT.get(iL, iL + sz);
			//std::cout << hR.val() << "__" << hL.val() << std::endl;
			//break;
			if (hR != hL)continue;
			int tmp = 1;
			int l = sz - i;
			if (l > 0)tmp += (l + n - 1) / n;
			int r = sz - (n - 1 - i);//???
			if (r > 0)tmp += (r + n - 1) / n;
			chmin(ans, tmp);
		}
		rep(i, n) {
			int sz = (m + 1) / 2;
			int iR = (i + 0) % n;
			mint hR = rhS.get(iR, iR + sz);
			int iL = (i - 1 + n) % n;
			iL = n - 1 - iL;
			mint hL = rhT.get(iL, iL + sz);
			if (hR != hL)continue;
			int tmp = 0 != i;
			int l = sz - i;
			int r = 0 == i ? sz : sz - (n - i);
			//cout << l << " " << r << endl;
			if (l > 0)tmp += (l + n - 1) / n;
			if (r > 0)tmp += (r + n - 1) / n;
			//cout << i << endl;
			chmin(ans, tmp);
		}
		if (LINF == ans)ans = -1;
		std::cout << ans << std::endl;
	}
	return 0;
}

yukicoder

結果

テストケース

ソースコード