这个是 $\textbf{2021.9.10}$ 未完成的最后一题。
就是先找找性质,发现 $f(0)$ 的时候本质上就是不包含 $0$ 的区间个数,不妨设 $0$ 在位置 $i$ 那么显然 $(i - 1) \times (n - i) = f(0)$。最多有两个这样的位置。
而且是对称的。
之后我们考虑进行 $\tt Dp$ 每次记录极大的区间也就是设 $f(i, l, r)$ 表示考虑了 $1 \sim i$ 的位置,极大区间是 $[l, r]$ 的方案数。
然后对于一个位置 $x$ 放在 $l$ 的左边的时候,可以满足 $c_i = (l - x + 1) \times (n - r + 1)$。
为了找到 $x$ 的位置显然 $n - r + 1 | c_i$。
如果出现 $c_i = 0$ 的情况,那么这个位置肯定只能放在 $[l, r]$ 中间,那么我们看一下是否能放即可。
直接讲一下优化,就是将之前的东西打表发现每一个 $dp$ 只有一个 $l$ 是有用的。我们不妨每次记录之前有用的 $l$ 然后用另外一个数组记录 $r$ 进行更新即可。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116
| #include <bits/stdc++.h> using namespace std;
#define Getmod
#ifdef Fread char buf[1 << 21], *iS, *iT; #define gc() (iS == iT ? (iT = (iS = buf) + fread (buf, 1, 1 << 21, stdin), (iS == iT ? EOF : *iS ++)) : *iS ++) #define getchar gc #endif
template <typename T> void r1(T &x) { x = 0; char c(getchar()); int f(1); for(; c < '0' || c > '9'; c = getchar()) if(c == '-') f = -1; for(; '0' <= c && c <= '9';c = getchar()) x = (x * 10) + (c ^ 48); x *= f; }
template <typename T,typename... Args> inline void r1(T& t, Args&... args) { r1(t); r1(args...); }
#ifdef Getmod const int mod = 998244353; template <int mod> struct typemod { int z; typemod(int a = 0) : z(a) {} inline int inc(int a,int b) const {return a += b - mod, a + ((a >> 31) & mod);} inline int dec(int a,int b) const {return a -= b, a + ((a >> 31) & mod);} inline int mul(int a,int b) const {return 1ll * a * b % mod;} typemod<mod> operator + (const typemod<mod> &x) const {return typemod(inc(z, x.z));} typemod<mod> operator - (const typemod<mod> &x) const {return typemod(dec(z, x.z));} typemod<mod> operator * (const typemod<mod> &x) const {return typemod(mul(z, x.z));} typemod<mod>& operator += (const typemod<mod> &x) {*this = *this + x; return *this;} typemod<mod>& operator -= (const typemod<mod> &x) {*this = *this - x; return *this;} typemod<mod>& operator *= (const typemod<mod> &x) {*this = *this * x; return *this;} int operator == (const typemod<mod> &x) const {return x.z == z;} int operator != (const typemod<mod> &x) const {return x.z != z;} }; typedef typemod<mod> Tm; #endif
const int maxn = 5e5 + 5; const int maxm = maxn << 1;
Tm f[2][maxn]; int st[2][maxn], n; typedef long long ll; ll c[maxn];
signed main() {
int i, j; r1(n); for(i = 1; i <= n; ++ i) r1(c[i - 1]); auto Calc = [&] (ll x) -> ll{ return x * (x + 1) >> 1; }; vector<int> ps; for(i = 1; i <= n; ++ i) { if(Calc(i - 1) + Calc(n - i) == c[0]) ps.push_back(i); } if(ps.empty()) return puts("0"), 0; f[0][ps[0]] = 1, st[0][ps[0]] = ps[0]; vector<ll> pre, nxt; pre.push_back(ps[0]);
vector<int> vis(n + 2, 0);
for(i = 1; i < n - 1; ++ i) { nxt.clear(); int bef((i - 1) & 1), now(i & 1); for(auto v : pre) { int l = v, r = st[bef][l]; if(vis[l]) continue; vis[l] = 1; if(!c[i]) { if(r - l + 1 - i > 0) { f[now][l] += f[bef][l] * (r - l + 1 - i); st[now][l] = r; nxt.push_back(l); } } else { ll L(l), R(n - r + 1); if(c[i] % R == 0) { ll nl = l - c[i] / R; f[now][nl] += f[bef][l]; st[now][nl] = r; nxt.push_back(nl); } if(c[i] % L == 0) { ll nr = r + c[i] / L; f[now][l] += f[bef][l]; st[now][l] = nr; nxt.push_back(l); } } } for(auto v : pre) vis[v] = 0, f[bef][v] = 0; nxt.swap(pre); } Tm ans(0); for(i = 1; i <= n; ++ i) ans += f[(n - 2) & 1][i]; ans *= Tm(ps.size()); printf("%d\n", ans.z); return 0; }
|