結果

問題 No.696 square1001 and Permutation 5
コンテスト
ユーザー square1001
提出日時 2018-05-30 21:11:19
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,615 ms / 10,000 ms
コード長 3,303 bytes
コンパイル時間 1,016 ms
コンパイル使用メモリ 82,512 KB
実行使用メモリ 62,736 KB
最終ジャッジ日時 2024-06-30 08:24:15
合計ジャッジ時間 9,020 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 12
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <string>
#include <vector>
#include <iostream>
#include <algorithm>
using namespace std;
const int mod = 998244353;
int powinv[29], roots[29];
void fourier_transform(int s, vector<int> &v, bool inverse) {
	if (v.size() != s) v.resize(s);
	for (int i = 0, j = 1; j < s - 1; ++j) {
		for (int k = s >> 1; k > (i ^= k); k >>= 1);
		if (i < j) swap(v[i], v[j]);
	}
	int sc = 0, sz = 1;
	while (sz < s) sz *= 2, ++sc;
	vector<int> pw(s + 1); pw[0] = 1;
	for (int i = 1; i <= s; i++) pw[i] = 1LL * pw[i - 1] * roots[sc] % mod;
	int qs = s;
	for (int b = 1; b < s; b <<= 1) {
		qs >>= 1;
		for (int i = 0; i < s; i += b * 2) {
			for (int j = i; j < i + b; ++j) {
				int f0 = v[j], f1 = 1LL * pw[(inverse ? b * 2 - j + i : j - i) * qs] * v[j + b] % mod;
				v[j] = (f0 + f1 < mod ? f0 + f1 : f0 + f1 - mod);
				v[j + b] = (f0 - f1 >= 0 ? f0 - f1 : f0 - f1 + mod);
			}
		}
	}
	if (inverse) {
		for (int i = 0; i < s; ++i) v[i] = 1LL * v[i] * powinv[sc] % mod;
	}
}
vector<int> convolve(vector<int> vx, vector<int> vy) {
	int s = max(vx.size(), vy.size()) * 2;
	fourier_transform(s, vx, false);
	fourier_transform(s, vy, false);
	for (int i = 0; i < s; ++i) vx[i] = 1LL * vx[i] * vy[i] % mod;
	fourier_transform(s, vx, true);
	int mx = 1;
	for (int i = 0; i < s; ++i) {
		if (vx[i] >= 100) {
			vx[i + 1] += vx[i] / 100;
			vx[i] %= 100;
		}
		while (mx < i + 1 && vx[i] > 0) mx *= 2;
	}
	vx.resize(mx);
	return vx;
}
int main() {
	cin.tie(0);
	ios_base::sync_with_stdio(false);
	powinv[1] = (mod + 1) / 2;
	for (int i = 2; i <= 23; ++i) powinv[i] = 1LL * powinv[i - 1] * powinv[1] % mod;
	roots[23] = 15311432;
	for (int i = 22; i >= 0; --i) roots[i] = 1LL * roots[i + 1] * roots[i + 1] % mod;
	int n;
	cin >> n;
	vector<int> p(n), bit(n);
	for (int i = 0; i < n; ++i) cin >> p[i];
	vector<vector<int> > a(n), b(n);
	for (int i = n - 1; i >= 0; --i) {
		int x = 0;
		for (int j = p[i] - 1; j >= 1; j -= j & (-j)) x += bit[j];
		for (int j = p[i]; j < n; j += j & (-j)) bit[j]++;
		int sa = 1, sb = 1, cb = i + 1;
		while (x > 0) {
			a[n - i - 1].push_back(x % 100); x /= 100;
			if (sa < a[n - i - 1].size()) sa *= 2;
		}
		a[n - i - 1].resize(sa);
		while (cb > 0) {
			b[i].push_back(cb % 100); cb /= 100;
			if (sb < b[i].size()) sb *= 2;
		}
		b[i].resize(sb);
	}
	int s = 1;
	while (2 << s < n) s++;
	for (int i = 0; i <= s; ++i) {
		for (int j = 0; j + (1 << i) < n; j += (2 << i)) {
			vector<int> c = convolve(b[j], a[j + (1 << i)]);
			b[j] = convolve(b[j], b[j + (1 << i)]);
			if(a[j].size() < c.size()) a[j].resize(c.size());
			for (int k = 0; k < c.size(); ++k) {
				a[j][k] += c[k];
				if (a[j][k] >= 100 && k + 1 != c.size()) a[j][k] -= 100, ++a[j][k + 1];
			}
			if (a[j].back() >= 100) {
				a[j].back() -= 100;
				a[j].resize(a[j].size() * 2);
				++a[j][a[j].size() >> 1];
			}
		}
	}
	++a[0][0];
	for (int i = 0; i + 1 < a[0].size(); ++i) {
		if (a[0][i] >= 100) a[0][i] -= 100, ++a[0][i + 1];
	}
	if (a[0].back() >= 100) {
		a[0].back() -= 100;
		a[0].resize(a[0].size() * 2);
		++a[0][a[0].size() >> 1];
	}
	string ret;
	for (int i = a[0].size() - 1; i >= 0; --i) ret += (a[0][i] < 10 ? "0" : "") + to_string(a[0][i]);
	bool flag = false;
	for (int i = 0; i < ret.size(); i++) {
		if (ret[i] != '0') flag = true;
		if (flag) cout << ret[i];
	}
	cout << '\n';
	return 0;
}
0