結果
問題 | No.1334 Multiply or Add |
ユーザー | phocom |
提出日時 | 2021-01-08 22:56:20 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 121 ms / 2,000 ms |
コード長 | 12,015 bytes |
コンパイル時間 | 2,020 ms |
コンパイル使用メモリ | 127,968 KB |
実行使用メモリ | 5,248 KB |
最終ジャッジ日時 | 2024-11-16 14:48:37 |
合計ジャッジ時間 | 4,533 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 71 |
ソースコード
#include <iostream>#include <vector>#include <set>#include <unordered_set>#include <map>#include <unordered_map>#include <cstdio>#include <bitset>#include <queue>#include <deque>#include <algorithm>#include <numeric>#include <cassert>#include <functional>#include <stack>#include <cmath>#include <string>#include <complex>#include <cassert>#define REP(i, N) for (int i = 0; i < (int)N; i++)#define FOR(i, a, b) for (int i = a; i < (int)b; i++)#define ALL(x) (x).begin(), (x).end()using namespace std;constexpr int inf = 1 << 30;constexpr long long llinf = 1LL << 62;constexpr int mod = 1000000007; // 998244353;constexpr int dy[4] = {-1, 0, 1, 0}, dx[4] = {0, -1, 0, 1};using ll = long long;using Pii = pair<int, int>;using Pll = pair<ll, ll>;inline int popcount(ll x) { return __builtin_popcountll(x); }inline int div2num(ll x) { return __builtin_ctzll(x); }inline bool bit(ll x, int b) { return (x >> b) & 1; }template <class T>bool chmax(T& a, const T& b) {if (a < b) {a = b;return 1;}return 0;}template <class T>bool chmin(T& a, const T& b) {if (b < a) {a = b;return 1;}return 0;}template <class T>string to_string(T s);template <class S, class T>string to_string(pair<S, T> p);string to_string(char c) { return string(1, c); }string to_string(string s) { return s; }string to_string(const char s[]) { return string(s); }template <class T>string to_string(T v) {if (v.empty()) return "{}";string ret = "{";for (auto x : v) ret += to_string(x) + ",";ret.back() = '}';return ret;}template <class S, class T>string to_string(pair<S, T> p) {return "{" + to_string(p.first) + ":" + to_string(p.second) + "}";}void debug() { cerr << endl; }template <class Head, class... Tail>void debug(Head head, Tail... tail) {cerr << to_string(head) << " ";debug(tail...);}template <class T>vector<T> makev(int n, T v) {return vector<T>(n, v);}struct IO {#ifdef _WIN32inline char getchar_unlocked() { return getchar(); }inline void putchar_unlocked(char c) { putchar(c); }#endifstd::string separator = " ";template <class T>inline void read(T& x) {char c;do {c = getchar_unlocked();} while (c != '-' && (c < '0' || '9' < c));bool minus = 0;if (c == '-') {minus = 1;c = getchar_unlocked();}x = 0;while ('0' <= c && c <= '9') {x *= 10;x += c - '0';c = getchar_unlocked();}if (minus) x = -x;}inline void read(std::string& x) {char c;do {c = getchar_unlocked();} while (c == ' ' || c == '\n');x.clear();do {x += c;c = getchar_unlocked();} while (c != ' ' && c != '\n' && c != EOF);}template <class T>inline void read(std::vector<T>& v) {for (auto& x : v) read(x);}template <class S, class T>inline void read(std::pair<S, T>& p) {read(p.first);read(p.second);}template <class Head, class... Tail>inline void read(Head& head, Tail&... tail) {read(head);read(tail...);}template <class T>inline void write(T x) {char buf[32];int p = 0;if (x < 0) {x = -x;putchar_unlocked('-');}if (x == 0) putchar_unlocked('0');while (x > 0) {buf[p++] = (x % 10) + '0';x /= 10;}while (p) {putchar_unlocked(buf[--p]);}}inline void write(std::string x) {for (char c : x) putchar_unlocked(c);}inline void write(const char s[]) {for (int i = 0; s[i] != 0; ++i) putchar_unlocked(s[i]);}template <class T>inline void write(std::vector<T> v) {if (v.empty()) return;for (auto itr = v.begin(); itr + 1 != v.end(); ++itr) {write(*itr);write(separator);}write(v.back());}template <class Head, class... Tail>inline void write(Head head, Tail... tail) {write(head);write(separator);write(tail...);}template <class Head, class... Tail>inline void writeln(Head head, Tail... tail) {write(head, tail...);write("\n");}void set_separator(std::string s) { separator = s; }} io;struct Prime {int n;vector<int> table;vector<int> primes;Prime(int _n = 100000) {n = _n + 1;table.resize(n, -1);table[0] = 0;table[1] = -1;for (int i = 2; i * i < n; ++i) {if (table[i] == -1) {for (int j = i * i; j < n; j += i) {table[j] = i;}}}}void enumerate_primes() {primes.clear();for (int i = 2; i < n; ++i) {if (table[i] == -1) primes.push_back(i);}}vector<pair<long long, int>> prime_factor(long long x) {map<long long, int> mp;long long div = 2;int p = 0;while (n <= x && div * div <= x) {if (x % div == 0) {mp[div]++;x /= div;} else {if (p + 1 < primes.size()) {div = primes[++p];} else {div++;}}}if (x < n) {while (table[x] != -1) {mp[table[x]]++;x /= table[x];}}if (x > 1) mp[x]++;vector<pair<long long, int>> ret;for (auto p : mp) ret.push_back(p);return ret;}};template <int MOD = 1000000007>struct Math {vector<long long> fact, factinv, inv;Math(int n = 100000) {fact.resize(n + 1);factinv.resize(n + 1);inv.resize(n + 1);fact[0] = fact[1] = 1;factinv[0] = factinv[1] = 1;inv[1] = 1;for (int i = 2; i <= n; ++i) {fact[i] = fact[i - 1] * i % MOD;inv[i] = MOD - inv[MOD % i] * (MOD / i) % MOD;factinv[i] = factinv[i - 1] * inv[i] % MOD;}}long long C(int n, int r) {if (n < r || n < 0 || r < 0) {return 0;} else {return fact[n] * (factinv[r] * factinv[n - r] % MOD) % MOD;}}long long P(int n, int r) {if (n < r || n < 0 || r < 0) {return 0;} else {return fact[n] * factinv[n - r] % MOD;}}long long H(int n, int r) { return C(n + r - 1, r); }};struct UnionFind {vector<int> data;vector<vector<int>> groups;UnionFind(int n) : data(n, -1) {}int root(int v) { return data[v] < 0 ? v : data[v] = root(data[v]); }bool unite(int u, int v) {if ((u = root(u)) == (v = root(v))) {return 1;} else {if (-data[u] < -data[v]) swap(u, v);data[u] += data[v];data[v] = u;return 0;}}int size(int v) { return -data[root(v)]; }void make_groups() {map<int, vector<int>> mp;for (int i = 0; i < data.size(); ++i) mp[root(i)].push_back(i);groups.clear();for (auto p : mp) groups.push_back(p.second);}};namespace phc {long long modpow(long long a, long long n) {long long res = 1;while (n > 0) {if (n & 1) res = res * a % mod;a = a * a % mod;n >>= 1;}return res;}long long modinv(long long a) {long long b = mod, u = 1, v = 0;while (b) {long long t = a / b;a -= t * b;swap(a, b);u -= t * v;swap(u, v);}u %= mod;if (u < 0) u += mod;return u;}long long gcd(long long a, long long b) { return b != 0 ? gcd(b, a % b) : a; }long long lcm(long long a, long long b) { return a / gcd(a, b) * b; }} // namespace phctemplate <int mod>struct ModInt {int x;ModInt() : x(0) {}ModInt(int64_t y) : x(y >= 0 ? y % mod : (mod - (-y) % mod) % mod) {}ModInt& operator+=(const ModInt& p) {if ((x += p.x) >= mod) x -= mod;return *this;}ModInt& operator-=(const ModInt& p) {if ((x += mod - p.x) >= mod) x -= mod;return *this;}ModInt& operator*=(const ModInt& p) {x = (int)(1LL * x * p.x % mod);return *this;}ModInt& operator/=(const ModInt& p) {*this *= p.inverse();return *this;}ModInt operator-() const { return ModInt(-x); }ModInt operator+(const ModInt& p) const { return ModInt(*this) += p; }ModInt operator-(const ModInt& p) const { return ModInt(*this) -= p; }ModInt operator*(const ModInt& p) const { return ModInt(*this) *= p; }ModInt operator/(const ModInt& p) const { return ModInt(*this) /= p; }ModInt inverse() const {int a = x, b = mod, u = 1, v = 0, t;while (b > 0) {t = a / b;swap(a -= t * b, b);swap(u -= t * v, v);}return ModInt(u);}ModInt pow(int64_t n) const {ModInt ret(1), mul(x);while (n > 0) {if (n & 1) ret *= mul;mul *= mul;n >>= 1;}return ret;}friend ostream& operator<<(ostream& os, const ModInt& p) { return os << p.x; }friend istream& operator>>(istream& is, ModInt& a) {int64_t t;is >> t;a = ModInt<mod>(t);return (is);}static int get_mod() { return mod; }};template <class T>struct RangeFold {int n;vector<T> dat;using F = function<T(T, T)>;const F f;const T id;RangeFold(int _n, const F _f, const T _id) : n(1), f(_f), id(_id) {while (n < _n) n <<= 1;dat.resize(2 * n, id);}void update(int i, T x) {dat[i += n] = x;while (i >>= 1) dat[i] = f(dat[i * 2], dat[i * 2 + 1]);}T query(int l, int r) {T xl = id, xr = id;for (l += n, r += n; l < r; l >>= 1, r >>= 1) {if (l & 1) xl = f(xl, dat[l++]);if (r & 1) xr = f(dat[--r], xr);}return f(xl, xr);}T& operator[](const int i) { return dat[n + i]; }};template <int L>struct KthNext {array<vector<int>, L> next;KthNext(int n, function<int(int)> f) {for (int i = 0; i < L; ++i) next[i].resize(n);for (int i = 0; i < n; ++i) next[0][i] = f(i);for (int i = 0; i < L - 1; ++i)for (int j = 0; j < n; ++j) next[i + 1][j] = next[i][next[i][j]];}int get(int x, long long k) {for (int i = 0; i < L; ++i)if ((k >> i) & 1) x = next[i][x];return x;}};long long inv_count(vector<int>& vec) {int n = vec.size();vector<int> bit(n + 1);long long ret = 0;for (int i = n - 1; i >= 0; --i) {int x = vec[i];while (x > 0) {ret += bit[x];x -= x & -x;}x = vec[i] + 1;while (x < n + 1) {bit[x] += 1;x += x & -x;}}return ret;}using modint = ModInt<mod>;int main() {ll N;io.read(N);vector<ll> A(N);io.read(A);if (*max_element(ALL(A)) == 1) {io.writeln(N);return 0;}int l = 0, r = N - 1;while (A[l] == 1) l++;while (A[r] == 1) r--;int f = 0;modint acc = 1;REP(i, N) {acc *= A[i];if (acc.x > 5e5) {f = 1;}}if (f) {acc += l + (N - 1 - r);io.writeln(acc.x);return 0;}vector<int> ls;int ct = 0;for (int i = l; i <= r; ++i) {if (A[i] > 1) {if (i > l) ls.push_back(ct);ls.push_back(A[i]);ct = 0;} else {ct++;}}int n = ls.size() / 2;ll ans = 0;REP(b, 1 << n) {vector<int> op;op.push_back(ls[0]);REP(i, n) {if (bit(b, i)) {op.push_back(-2);op.push_back(ls[i * 2 + 2]);} else {op.push_back(-1);op.push_back(ls[i * 2 + 1]);op.push_back(-1);op.push_back(ls[i * 2 + 2]);}}queue<int> q;stack<int> st;for (int x : op) {if (x >= 0) {q.push(x);} else if (x == -1) {while (!st.empty() && st.top() == -2) {st.pop();q.push(-2);}st.push(x);} else {st.push(x);}}while (!st.empty()) {q.push(st.top());st.pop();}while (!q.empty()) {int x = q.front();q.pop();if (x >= 0) {st.push(x);} else if (x == -1) {int v = st.top();st.pop();int u = st.top();st.pop();st.push(v + u);} else {int v = st.top();st.pop();int u = st.top();st.pop();st.push(v * u);}}chmax(ans, st.top() + l + (N - 1 - r));}io.writeln(ans);return 0;}