結果

問題 No.599 回文かい
ユーザー cutmdo
提出日時 2019-08-24 06:44:34
言語 C++17(1z)
(gcc 8.3.0)
結果
AC  
実行時間 195 ms
コード長 4,134 Byte
コンパイル時間 893 ms
使用メモリ 1,680 KB
最終ジャッジ日時 2019-10-07 19:41:42

テストケース

テストケース表示
入力 結果 実行時間
使用メモリ
0_0.txt AC 3 ms
1,524 KB
0_1.txt AC 3 ms
1,524 KB
0_2.txt AC 3 ms
1,536 KB
0_3.txt AC 3 ms
1,524 KB
1_0.txt AC 3 ms
1,544 KB
1_1.txt AC 3 ms
1,544 KB
1_2.txt AC 4 ms
1,544 KB
1_3.txt AC 4 ms
1,548 KB
1_4.txt AC 3 ms
1,540 KB
1_5.txt AC 3 ms
1,544 KB
2_0.txt AC 35 ms
1,672 KB
2_1.txt AC 21 ms
1,664 KB
2_2.txt AC 38 ms
1,672 KB
2_3.txt AC 25 ms
1,668 KB
2_4.txt AC 62 ms
1,672 KB
2_5.txt AC 83 ms
1,680 KB
2_6.txt AC 168 ms
1,676 KB
2_7.txt AC 195 ms
1,680 KB
3_0.txt AC 2 ms
1,544 KB
3_1.txt AC 3 ms
1,544 KB
3_2.txt AC 3 ms
1,540 KB
evil_0.txt AC 52 ms
1,676 KB
テストケース一括ダウンロード

ソースコード

diff #
#include <iostream>
#include <iomanip>
#include <string>
#include <cmath>
#include <algorithm>
#include <vector>
#include <set>
#include <map>
#include <unordered_map>
#include <unordered_set>
#include <list>
#include <stack>
#include <queue>
#include <bitset>
#include <numeric>
#include <cassert>
#ifdef DEBUG
#include "./Lib/debug.hpp"
#else
#define dump(...)
template<class T>inline auto d_val(T a, T b) { return a; }
#endif

/* (=^o^=) */
#define int ll

/* macro */
#define FOR(i, b, e) for(ll i = (ll)(b); i < (ll)(e); ++i)
#define RFOR(i, b, e) for(ll i = (ll)(e-1); i >= (ll)(b); --i)
#define REP(i, n) FOR(i, 0, n)
#define RREP(i, n) RFOR(i, 0, n)
#define REPC(x,c) for(const auto& x:(c))
#define REPI2(it,b,e) for(auto it = (b); it != (e); ++it)
#define REPI(it,c) REPI2(it, (c).begin(), (c).end())
#define RREPI(it,c) REPI2(it, (c).rbegin(), (c).rend())
#define REPI_ERACE2(it, b, e) for(auto it = (b); it != (e);)
#define REPI_ERACE(it, c) REPI_ERACE2(it, (c).begin(), (c).end())
#define ALL(x) (x).begin(),(x).end()
#define cauto const auto&
/* macro func */
#define SORT(x) do{sort(ALL(x));}while(false)
#define RSORT(x) do{sort((x).rbegin(),(x).rend());}while(false)
#define UNIQUE(v) do{v.erase( unique(v.begin(), v.end()), v.end() );}while(false)
#define MAX(x,y) do{x = std::max(x,y);}while(false)
#define MIN(x,y) do{x = std::min(x,y);}while(false)
#define BR do{cout<<"\n";}while(false)

/* type define */
using ll = long long;
using PAIR = std::pair<ll, ll>;
using VS = std::vector<std::string>;
using VL = std::vector<long long>;
using VVL = std::vector<VL>;
using VVVL = std::vector<VVL>;
using VD = std::vector<double>;
template<class T>
using V = std::vector<T>;

/* using std */
using std::cout;
constexpr char endl = '\n';
using std::cin;
using std::sort;
using std::pair;
using std::string;
using std::stack;
using std::queue;
using std::vector;
using std::list;
using std::map;
using std::unordered_map;
using std::multimap;
using std::unordered_multimap;
using std::set;
using std::unordered_set;
using std::unordered_multiset;
using std::multiset;
using std::bitset;
using std::priority_queue;

/* constant value */
constexpr ll MOD = 1000000007;
//constexpr ll MOD = 998244353;

/* Initial processing  */
struct Preprocessing { Preprocessing() { std::cin.tie(0); std::ios::sync_with_stdio(0); }; }_Preprocessing;

/* Remove the source of the bug */
signed pow(signed, signed) { assert(false); return -1; }

/* define hash */
namespace std { template <>	class hash<std::pair<ll, ll>> { public:	size_t operator()(const std::pair<ll, ll>& x) const { return hash<ll>()(1000000000 * x.first + x.second); } }; }

/* input */
template<class T> std::istream& operator >> (std::istream& is, vector<T>& vec) { for (T& x : vec) is >> x; return is; }

//=============================================================================================

class ZAlgorithm {
	const std::vector<int> m_zArray;

	static auto constrcutZArray(const std::string& str) {
		auto sz = str.size();
		auto zArray = std::vector<int>(sz);
		zArray[0] = sz;

		int i = 1, j = 0;
		while (i < sz) {
			while (i + j < sz && str[j] == str[i + j]) ++j;
			zArray[i] = j;
			if (j == 0) { ++i; continue; }
			int k = 1;
			while (i + k < sz && k + zArray[k] < j) {
				zArray[i + k] = zArray[k];
				++k;
			}
			i += k; j -= k;
		}
		return zArray;
	}
public:
	ZAlgorithm(const std::string& str) :m_zArray(constrcutZArray(str)) {}

	auto getZArray() const {
		return m_zArray;
	}

	/* output fot debug */
	void debugOutput() const {
		dump(m_zArray);
	}

};

signed main() {
	string str;
	cin >> str;

	auto sz = static_cast<ll>(str.size());

	auto o = (!(sz & 1));
	VL dp((sz + 1) / 2 + o, 1);

	REP(i, (sz + 1) / 2) {
		auto s = str.substr((sz + 1) / 2 - i - 1, 2 * i + 1 + o);
		auto z = ZAlgorithm(s).getZArray();

		//cout << "-- " << s << " --" << endl;
		auto ssz = static_cast<ll>(s.size());
		REP(j, s.size() / 2) {
			if (z[ssz - (j + 1)] == j + 1) {
				auto p = (ssz - 1) / 2 - j - 1;
				if (p + o >= 0) {
					dp[i + o] += dp[p + o];
					dp[i + o] %= MOD;
				}
			};
		}
	}
	dump(dp);
	cout << *dp.rbegin() << endl;
}
0