#include <vector> #include <cstdio> #include <stack> #include <algorithm> #include <queue> #include <map> #include <cmath> #include <unordered_set> #include <unordered_map> #include <set> #include <cstring> #include <bitset> #include <iostream> #include <queue> #include <iomanip> #include <complex> using namespace std; #define ALL(x) x.begin(), x.end() #define UNIQUE(c) (c).resize(unique(ALL(c)) - (c).begin()) #define FOR(i, a, b) for(int i =(a); i <=(b); ++i) #define RE(i, n) FOR(i, 1, n) #define RED(i, n) FORD(i, n, 1) #define FORD(i, a, b) for(int i = (a); i >= (b); --i) #define REP(i, n) for(int i = 0;i <(n); ++i) #define REPD(i, n) FORD(i, n-1,0) #define printA(a, L, R) FOR(i,L,R) cout << a[i] << (i==R?'\n':' ') #define printV(a) printA(a,0,a.size()-1) #define PRESENT(c, x) ((c).find(x) != (c).end()) #define CPRESENT(c, x) (find(ALL(c),x) != (c).end()) #define bit(x, i) (x&(1<<i)) #define lowbit(x) ((x)&((x)^((x)-1))) const int fx[4][2] = { {0, 1}, {1, 0}, {0, -1}, {-1, 0} }; const int fxx[8][2] = { {0, 1}, {0, -1}, {1, 0}, {-1, 0}, {1, 1}, {1, -1}, {-1, 1}, {-1, -1} }; const int sk[8][2] = { {2, 1}, {2, -1}, {-2, -1}, {-2, 1}, {1, 2}, {-1, 2}, {1, -2}, {-1, -2} }; const int maxn = 200012; long long mod = 998244353ll; const long long inf = 10000000000012ll; const long long mam = 200; long long n, l, r, mid, a[maxn]; void solve() { cin >> n; REP(i, n) cin >> a[i]; l = 3; r = n * mam; while (l <= r) { mid = (l + r) / 2; bool ok = true; long long akt = mid, lewo = 0ll; REP(i, n) { long long l2 = 0, r2 = min(mam, akt); while (l2 <= r2) { long long mid2 = (l2 + r2) / 2; long long res = mid2 * (mid2 - 1ll) * (mid2 - 2ll) / 6ll; long long prawo = (i < n - 1) ? (akt - mid2) : 0ll; res += (lewo + prawo) * mid2 * (mid2 - 1ll) / 2ll; res += lewo * prawo * mid2; if (res < a[i])l2 = mid2 + 1ll; else r2 = mid2 - 1ll; } if (l2 > akt) { ok = false; break; } akt -= l2; lewo += l2; } if (ok) r = mid - 1; else l = mid + 1; } cout<<l<<"\n"; } int main() { ios_base::sync_with_stdio(false), cin.tie(nullptr); int tt = 1; cin >> tt; while (tt--) { solve(); } return 0; }
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 | #include <vector> #include <cstdio> #include <stack> #include <algorithm> #include <queue> #include <map> #include <cmath> #include <unordered_set> #include <unordered_map> #include <set> #include <cstring> #include <bitset> #include <iostream> #include <queue> #include <iomanip> #include <complex> using namespace std; #define ALL(x) x.begin(), x.end() #define UNIQUE(c) (c).resize(unique(ALL(c)) - (c).begin()) #define FOR(i, a, b) for(int i =(a); i <=(b); ++i) #define RE(i, n) FOR(i, 1, n) #define RED(i, n) FORD(i, n, 1) #define FORD(i, a, b) for(int i = (a); i >= (b); --i) #define REP(i, n) for(int i = 0;i <(n); ++i) #define REPD(i, n) FORD(i, n-1,0) #define printA(a, L, R) FOR(i,L,R) cout << a[i] << (i==R?'\n':' ') #define printV(a) printA(a,0,a.size()-1) #define PRESENT(c, x) ((c).find(x) != (c).end()) #define CPRESENT(c, x) (find(ALL(c),x) != (c).end()) #define bit(x, i) (x&(1<<i)) #define lowbit(x) ((x)&((x)^((x)-1))) const int fx[4][2] = { {0, 1}, {1, 0}, {0, -1}, {-1, 0} }; const int fxx[8][2] = { {0, 1}, {0, -1}, {1, 0}, {-1, 0}, {1, 1}, {1, -1}, {-1, 1}, {-1, -1} }; const int sk[8][2] = { {2, 1}, {2, -1}, {-2, -1}, {-2, 1}, {1, 2}, {-1, 2}, {1, -2}, {-1, -2} }; const int maxn = 200012; long long mod = 998244353ll; const long long inf = 10000000000012ll; const long long mam = 200; long long n, l, r, mid, a[maxn]; void solve() { cin >> n; REP(i, n) cin >> a[i]; l = 3; r = n * mam; while (l <= r) { mid = (l + r) / 2; bool ok = true; long long akt = mid, lewo = 0ll; REP(i, n) { long long l2 = 0, r2 = min(mam, akt); while (l2 <= r2) { long long mid2 = (l2 + r2) / 2; long long res = mid2 * (mid2 - 1ll) * (mid2 - 2ll) / 6ll; long long prawo = (i < n - 1) ? (akt - mid2) : 0ll; res += (lewo + prawo) * mid2 * (mid2 - 1ll) / 2ll; res += lewo * prawo * mid2; if (res < a[i])l2 = mid2 + 1ll; else r2 = mid2 - 1ll; } if (l2 > akt) { ok = false; break; } akt -= l2; lewo += l2; } if (ok) r = mid - 1; else l = mid + 1; } cout<<l<<"\n"; } int main() { ios_base::sync_with_stdio(false), cin.tie(nullptr); int tt = 1; cin >> tt; while (tt--) { solve(); } return 0; } |