QOJ.ac
QOJ
ID | Submission ID | Problem | Hacker | Owner | Result | Submit time | Judge time |
---|---|---|---|---|---|---|---|
#1235 | #426021 | #7219. The Mighty Spell | ASnown | caijianhong | Failed. | 2024-11-22 15:02:52 | 2024-11-22 15:02:52 |
Details
Extra Test:
Accepted
time: 0ms
memory: 6700kb
input:
60 3 2 3 2 2 3 1 3 1 2 3 2 2 3 1 2 2 2 1 2 3 1 3 3 2 2 2 3 3 1 2 1 1 2 1 1 1 1 1 1 1 3 2 1 2 3 3 3 1 2 2 2 2 3 3 1 1 3 2 2 2
output:
628462825
result:
ok answer is '628462825'
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#426021 | #7219. The Mighty Spell | caijianhong | AC ✓ | 642ms | 49560kb | C++14 | 3.8kb | 2024-05-30 20:21:43 | 2024-11-22 16:49:49 |
answer
#include <bits/stdc++.h>
using namespace std;
#ifdef LOCAL
#define debug(...) fprintf(stderr, __VA_ARGS__)
#else
#define endl "\n"
#define debug(...) void(0)
#endif
using LL = long long;
template <class T>
using must_int = enable_if_t<is_integral<T>::value, int>;
template <unsigned umod>
struct modint {
static constexpr int mod = umod;
unsigned v;
modint() : v(0) {}
template <class T, must_int<T> = 0>
modint(T x) {
x %= mod;
v = x < 0 ? x + mod : x;
}
modint operator+() const { return *this; }
modint operator-() const { return modint() - *this; }
friend int raw(const modint &self) { return self.v; }
friend ostream& operator<<(ostream& os, const modint &self) {
return os << raw(self);
}
modint &operator+=(const modint &rhs) {
v += rhs.v;
if (v >= umod) v -= umod;
return *this;
}
modint &operator-=(const modint &rhs) {
v -= rhs.v;
if (v >= umod) v += umod;
return *this;
}
modint &operator*=(const modint &rhs) {
v = 1ull * v * rhs.v % umod;
return *this;
}
modint &operator/=(const modint &rhs) {
assert(rhs.v);
return *this *= qpow(rhs, mod - 2);
}
template <class T, must_int<T> = 0>
friend modint qpow(modint a, T b) {
modint r = 1;
for (; b; b >>= 1, a *= a)
if (b & 1) r *= a;
return r;
}
friend modint operator+(modint lhs, const modint &rhs) { return lhs += rhs; }
friend modint operator-(modint lhs, const modint &rhs) { return lhs -= rhs; }
friend modint operator*(modint lhs, const modint &rhs) { return lhs *= rhs; }
friend modint operator/(modint lhs, const modint &rhs) { return lhs /= rhs; }
bool operator==(const modint &rhs) const { return v == rhs.v; }
bool operator!=(const modint &rhs) const { return v != rhs.v; }
};
#ifdef ONLINE_JUDGE
typedef modint<(int)1e9 + 7> mint;
#else
typedef modint<998244353> mint;
#endif
int n, m, a[200010], cnt[60][200010];
mint f[200010], b[200010], qp2[200010];
mint g(mint l) { return l != 0 ? 2 * l * l * l + 3 * l * l + 3 * l + 3 : 0; }
void init() {
f[1] = g(1);
mint pre = f[1];
for (int i = 2; i <= n; i++) {
f[i] = g(i) - g(i - 1) - pre;
pre += f[i];
}
for (int c = 1; c <= m; c++) {
for (int i = 1; i <= n; i++) cnt[c][i] = cnt[c][i - 1] + (a[i] == c);
}
qp2[0] = 1;
for (int i = 1; i <= n; i++) qp2[i] = qp2[i - 1] * 2;
}
int calc(int l, int r, int c) {
return cnt[c][n] - cnt[c][r] + cnt[c][l - 1];
}
mint solve() {
vector<tuple<int, LL, mint, mint>> vec;
for (int l = n; l >= 1; l--) {
decltype(vec) nxt;
int cntl = cnt[a[l]][n];
{
mint tmp = 1;
for (int c = 1; c <= m; c++) if (c != a[l]) tmp *= qp2[calc(l, l, c)] - 1;
nxt.emplace_back(l, 1ll << a[l], qp2[cntl], tmp);
}
LL lstmask = 1ll << a[l];
for (auto [r, mask, ci, co] : vec) {
if (lstmask == mask) continue;
if (mask >> a[l] & 1) {
nxt.emplace_back(r, mask, ci, co);
lstmask = mask;
} else {
nxt.emplace_back(r, mask | 1ll << a[l], ci * qp2[cntl], co / (qp2[cntl] - 1));
lstmask = mask | 1ll << a[l];
}
}
vec = move(nxt);
for (int j = 0; j < (int)vec.size(); j++) {
int rn = j + 1 < (int)vec.size() ? get<0>(vec[j + 1]) : n + 1;
auto [r, mask, ci, co] = vec[j];
b[r - l + 1] += ci * co, b[rn - l + 1] -= ci * co;
}
}
mint ans = 0;
for (int i = 1; i <= n; i++) b[i] += b[i - 1], ans += b[i] * f[i] / qp2[i];
return ans;
}
int main() {
#ifndef LOCAL
cin.tie(nullptr)->sync_with_stdio(false);
#endif
cin >> n >> m;
for (int i = 1; i <= n; i++) cin >> a[i];
if (set<int>(a + 1, a + n + 1).size() < m) {
cout << 0 << endl;
return 0;
}
init();
cout << solve() << endl;
return 0;
}