#717028 (C++14) No.1334 Multiply or Add

提出ソース

結果

問題	No.1334 Multiply or Add
ユーザー	XD
提出日時	2021-11-17 16:12:49
言語	C++14 (gcc 13.2.0 + boost 1.83.0)
結果	AC
実行時間	15 ms / 2,000 ms
コード長	4,230 bytes
コンパイル時間	1,676 ms
コンパイル使用メモリ	167,232 KB
実行使用メモリ	8,408 KB
最終ジャッジ日時	2023-08-25 04:19:56
合計ジャッジ時間	5,208 ms
ジャッジサーバーID （参考情報）	judge15 / judge12

このコードへのチャレンジ（β）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 5,456 KB
testcase_01	AC	2 ms 5,444 KB
testcase_02	AC	2 ms 5,388 KB
testcase_03	AC	4 ms 8,284 KB
testcase_04	AC	4 ms 8,288 KB
testcase_05	AC	4 ms 8,272 KB
testcase_06	AC	4 ms 8,180 KB
testcase_07	AC	4 ms 8,272 KB
testcase_08	AC	4 ms 8,168 KB
testcase_09	AC	4 ms 8,276 KB
testcase_10	AC	4 ms 8,172 KB
testcase_11	AC	5 ms 8,180 KB
testcase_12	AC	4 ms 8,236 KB
testcase_13	AC	4 ms 8,340 KB
testcase_14	AC	5 ms 8,216 KB
testcase_15	AC	4 ms 8,216 KB
testcase_16	AC	5 ms 8,240 KB
testcase_17	AC	5 ms 8,240 KB
testcase_18	AC	5 ms 8,280 KB
testcase_19	AC	15 ms 8,176 KB
testcase_20	AC	15 ms 8,344 KB
testcase_21	AC	15 ms 8,284 KB
testcase_22	AC	5 ms 8,264 KB
testcase_23	AC	4 ms 8,172 KB
testcase_24	AC	4 ms 8,332 KB
testcase_25	AC	4 ms 8,288 KB
testcase_26	AC	4 ms 8,288 KB
testcase_27	AC	5 ms 8,176 KB
testcase_28	AC	4 ms 8,172 KB
testcase_29	AC	4 ms 8,216 KB
testcase_30	AC	4 ms 8,172 KB
testcase_31	AC	5 ms 8,348 KB
testcase_32	AC	4 ms 8,180 KB
testcase_33	AC	4 ms 8,236 KB
testcase_34	AC	4 ms 8,280 KB
testcase_35	AC	4 ms 8,276 KB
testcase_36	AC	4 ms 8,172 KB
testcase_37	AC	4 ms 8,272 KB
testcase_38	AC	4 ms 8,172 KB
testcase_39	AC	2 ms 5,408 KB
testcase_40	AC	3 ms 5,832 KB
testcase_41	AC	2 ms 5,532 KB
testcase_42	AC	3 ms 5,524 KB
testcase_43	AC	4 ms 8,196 KB
testcase_44	AC	2 ms 5,712 KB
testcase_45	AC	2 ms 5,468 KB
testcase_46	AC	3 ms 5,772 KB
testcase_47	AC	4 ms 8,048 KB
testcase_48	AC	3 ms 5,752 KB
testcase_49	AC	4 ms 8,160 KB
testcase_50	AC	8 ms 8,376 KB
testcase_51	AC	8 ms 8,172 KB
testcase_52	AC	8 ms 8,276 KB
testcase_53	AC	8 ms 8,284 KB
testcase_54	AC	8 ms 8,232 KB
testcase_55	AC	8 ms 8,308 KB
testcase_56	AC	2 ms 5,464 KB
testcase_57	AC	2 ms 5,480 KB
testcase_58	AC	2 ms 5,392 KB
testcase_59	AC	2 ms 5,396 KB
testcase_60	AC	2 ms 5,392 KB
testcase_61	AC	2 ms 5,356 KB
testcase_62	AC	2 ms 5,512 KB
testcase_63	AC	2 ms 5,444 KB
testcase_64	AC	2 ms 5,392 KB
testcase_65	AC	2 ms 5,400 KB
testcase_66	AC	7 ms 8,140 KB
testcase_67	AC	5 ms 7,816 KB
testcase_68	AC	2 ms 5,468 KB
testcase_69	AC	7 ms 8,408 KB
testcase_70	AC	7 ms 8,132 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <bits/stdc++.h>
#define forn(i,s,t) for(register int i=(s); i<=(t); ++i)
#define forl(i,s,t) for(register i64 i=(s); i<=(t); ++i)
#define form(i,s,t) for(register int i=(s); i>=(t); --i)
#define rep(i,s,t) for(register int i=(s); i<(t); ++i)
#define IT(u) for(register int i=G[u]; i; i=E[i].nxt)
using namespace std;
namespace FASTIO {
	const int SIZ = 1 << 26 | 1;
	char ibuf[SIZ], obuf[SIZ], *iS = ibuf, *iT = ibuf, *oS = obuf, *oT = obuf + SIZ - 1, qwq[60], qaq;
	#define gc() (iS == iT && (iT = (iS = ibuf) + fread(ibuf, 1, SIZ, stdin), iS == iT) ? EOF : *iS++)
	template<typename T>
	inline void Rdn(T& A) {
		register bool fl = 0;
		register char ch = gc(); A = 0;
		while(!isdigit(ch)) fl = (ch == '-'), ch = gc();
		while(isdigit(ch)) A = (A * 10) + (ch & 15), ch = gc();
		fl && (A = -A);
	}
	inline void Rdn(char& c) {while((c = gc()) == ' ' || c == '\n' || c == '\r');}
	inline void Rdn(char* s) {
		while((*s = gc()) == ' ' || *s == '\n' || *s == '\r') ;
		if(*s == EOF) return ;
		while(*s != ' ' && *s != '\n' && *s != '\r' && *s != EOF) *(++s) = gc();
		*s = 0;
	}
	template<typename T, typename ...U>
	inline void Rdn(T& A, U& ...B) {Rdn(A), Rdn(B...);}
	inline void flush() {fwrite(obuf, 1, oS - obuf, stdout), oS = obuf;}
	inline void pc(char c) {*oS ++ = c; if(oS == oT) flush();}
	template<typename T>
	inline void Wtn(T A) {
		if(!A) return pc('0'); if(A < 0) pc('-'), A = -A;
		while(A) qwq[++qaq] = A % 10 + '0', A /= 10;
		while(qaq) pc(qwq[qaq -- ]);
	}
	inline void Wtn(char A) {pc(A);}
	inline void Wtn(char *s) {while(*s) pc(*s), ++s;}
	inline void Wtn(const char *s) {while(*s) pc(*s), ++s;}
	template<typename T, typename ...U>
	inline void Wtn(T A, U ...B) {Wtn(A), Wtn(B...);}
#undef gc
}
using FASTIO :: Rdn;
using FASTIO :: Wtn;
using FASTIO :: flush;
const int Mod = 1e9 + 7;
namespace Modint {
	struct Mint {
		int res;
		Mint() {}
		Mint(int _r) : res(_r) {}
		inline friend Mint operator + (const Mint& A, const Mint& B) {
			return Mint((A.res + B.res >= Mod) ? (A.res + B.res - Mod) : (A.res + B.res));
		}
		inline friend Mint operator - (const Mint& A, const Mint& B) {return A + Mint(Mod - B.res); }
		inline friend Mint operator * (const Mint& A, const Mint& B) {return Mint(1ll * A.res * B.res % Mod); }
		inline friend Mint& operator += (Mint& A, const Mint& B) {return A = A + B; }
		inline friend Mint& operator -= (Mint& A, const Mint& B) {return A = A - B; }
		inline friend Mint& operator *= (Mint& A, const Mint& B) {return A = A * B; }
		inline friend Mint q_pow(Mint p, int k = Mod - 2) {
			Mint res(1);
			for (; k; k >>= 1, p *= p) (k & 1) && (res *= p, 0);
			return res;
		}
	} ;
}
using Modint :: Mint;
typedef long long i64;
typedef double f64;
typedef unsigned long long u64;
typedef pair<i64, i64> pii;
typedef pair<int, u64> piu;
const int N = 2e5 + 5;
const i64 INF = 1e18;
inline void init() {}
struct node {
	int l, r, val;
	node() {}
	node(int _l, int _r, int _v) : l(_l), r(_r), val(_v) {}
} b[N];
int n, a[N], m, non[N], s;
inline void solve() {
	Rdn(n);
	int Prod = 1; bool GG = 0;
	forn (i, 1, n) {
		Rdn(a[i]); 
		if (GG || 1ll * Prod * a[i] > 2 * n) GG = 1;
		else Prod *= a[i];
	}
	if (GG) {
		int now = 1; Mint res(0), Prd(1);
		while (now <= n && a[now] == 1) res += Mint(1), ++now;
		int nnw = n;
		while (nnw > now && a[nnw] == 1) res += Mint(1), nnw--;
		forn (i, now, nnw) Prd *= Mint(a[i]);
		res += Prd;
		Wtn(res.res, '\n');
		return ;
	}
	int Rdx = 0;
	forn (l, 1, n) {
		int r = l; int tmp(a[l]);
		while (r < n && (a[r + 1] > 1) == (a[l] > 1)) ++r, tmp *= a[r];
		b[++m] = node(l, r, tmp);
//		Wtn("[", l, ", ", r, "]\n");
		if (l == 1 && tmp == 1) --m, Rdx += r - l + 1;
		else if (r == n && tmp == 1) --m, Rdx += r - l + 1;
		else if (tmp == 1) non[s++] = m;
		l = r;
	}
	int ans = 0; b[0] = b[m + 1] = node(0, 0, 1);
	rep (S, 0, 1 << s) {
		int res = (b[1].val != 1) * b[1].val, lst = (b[1].val != 1) * b[1].val;
		rep (i, 0, s) {
			if (S >> i & 1) res -= lst, lst *= b[non[i] + 1].val, res += lst;
			else res += b[non[i]].r - b[non[i]].l + 1 + b[non[i] + 1].val, lst = b[non[i] + 1].val;
		}
		ans = max(ans, res);
	}
	Wtn(ans + Rdx, '\n');
}
int Trd;
int main() {
	Trd = 1;
	while(Trd--) init(), solve(); flush();
	return 0;
}

yukicoder

結果

テストケース

ソースコード