結果

問題 No.3247 Multiplication 8 2
ユーザー 遭難者
提出日時 2025-07-09 13:12:37
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 3,652 bytes
コンパイル時間 4,364 ms
コンパイル使用メモリ 264,308 KB
実行使用メモリ 49,548 KB
最終ジャッジ日時 2025-07-09 13:34:51
合計ジャッジ時間 15,590 ms
ジャッジサーバーID
(参考情報)
judge3 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 23 WA * 5
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;
using std::cerr;
using std::cin;
using std::cout;
#if __has_include(<atcoder/all>)
#include <atcoder/all>
using mint = atcoder::modint998244353;
istream &operator>>(istream &is, mint &a) {
	int t;
	is >> t;
	a = t;
	return is;
}
ostream &operator<<(ostream &os, mint a) { return os << a.val(); }
#endif
typedef long double ld;
#define long long long
#define uint unsigned int
#define ull unsigned long
#define overload3(a, b, c, name, ...) name
#define rep3(i, a, b) for (int i = (a); i < (b); i++)
#define rep2(i, n) rep3(i, 0, n)
#define rep1(n) rep2(i, n)
#define rep(...) overload3(__VA_ARGS__, rep3, rep2, rep1)(__VA_ARGS__)
#define per3(i, a, b) for (int i = (b) - 1; i >= (a); i--)
#define per2(i, n) per3(i, 0, n)
#define per1(n) per2(i, n)
#define per(...) overload3(__VA_ARGS__, per3, per2, per1)(__VA_ARGS__)
#define all(a) a.begin(), a.end()
#define UNIQUE(a)                                                              \
	sort(all(a));                                                              \
	a.erase(unique(all(a)), a.end())
#define sz(a) (int)a.size()
#define vec vector
#ifndef DEBUG
#define cerr                                                                   \
	if (0)                                                                     \
	cerr
// #undef assert
// #define assert(...) void(0)
#undef endl
#define endl '\n'
#endif
template <typename T> ostream &operator<<(ostream &os, vector<T> a) {
	const int n = a.size();
	rep(i, n) {
		os << a[i];
		if (i + 1 != n)
			os << " ";
	}
	return os;
}
template <typename T, size_t n>
ostream &operator<<(ostream &os, array<T, n> a) {
	rep(i, n) {
		os << a[i];
		if (i + 1 != n)
			os << " ";
	}
	return os;
}
template <typename T> istream &operator>>(istream &is, vector<T> &a) {
	for (T &i : a)
		is >> i;
	return is;
}
template <typename T, typename S> bool chmin(T &x, S y) {
	if ((T)y < x) {
		x = (T)y;
		return true;
	}
	return false;
}
template <typename T, typename S> bool chmax(T &x, S y) {
	if (x < (T)y) {
		x = (T)y;
		return true;
	}
	return false;
}
template <typename T> void operator++(vector<T> &a) {
	for (T &i : a)
		++i;
}
template <typename T> void operator--(vector<T> &a) {
	for (T &i : a)
		--i;
}
template <typename T> void operator++(vector<T> &a, int) {
	for (T &i : a)
		i++;
}
template <typename T> void operator--(vector<T> &a, int) {
	for (T &i : a)
		i--;
}
void solve() {
	int n, po;
	cin >> n >> po;
	vec<int> a(n);
	cin >> a;
	vec<vec<int>> g(n);
	g[0].push_back(-1);
	int tw = 0;
	bool fu = true;
	rep(i, n) {
		if (a[i] < 0) {
			fu ^= true;
			a[i] *= -1;
		}
		if (a[i] == 2)
			tw += 1;
		if (tw >= 3 && tw % 3 == 0 && fu) {
			g[tw / 3].push_back(i);
		}
	}
	if (tw % 3 != 0 || !fu) {
		cout << 0 << endl;
		return;
	}
	const int m = tw / 3 + 1;
	rep(i, m) if (g[i].empty()) {
		cout << 0 << endl;
		return;
	}
	g.resize(m);
	g[m - 1].clear();
	g[m - 1].push_back(n - 1);
	mint pr = 1;
	for (auto &gg : g)
		pr *= sz(gg);
	mint ans = 0;
	rep(i, 1, m) {
		auto gg1 = g[i - 1];
		auto gg2 = g[i];
		vec<mint> a(gg1[sz(gg1) - 1] - gg1[0] + 1);
		vec<mint> b(gg2[sz(gg2) - 1] - gg2[0] + 1);
		for (int v : gg1)
			a[v - gg1[0]]++;
		for (int v : gg2)
			b[gg2[sz(gg2) - 1] - v]++;
		auto c = atcoder::convolution(a, b);
		mint sinv = mint(sz(gg1) * sz(gg2)).inv();
		rep(j, sz(c)) {
			mint le = gg2[sz(gg2) - 1] - gg1[0] - j;
			ans += c[j] * le.pow(po) * pr * sinv;
		}
	}
	cout << ans << endl;
	return;
}
int main() {
	// srand((unsigned)time(NULL));
	cin.tie(nullptr);
	ios::sync_with_stdio(false);
	cout << fixed << setprecision(20);
	int t = 1;
	// cin >> t;
	while (t--)
		solve();
}
0