#include <bits/stdc++.h> using namespace std; #define ll long long #define ld long double #define all(a) (a).begin(), (a).end() #define allr(a) (a).rbegin(), (a).rend() #define F first #define S second #define pb push_back typedef pair<int, int> pii; typedef pair<ll, ll> pll; /* #ifndef ONLINE_JUDGE #include <cpp-dump.hpp> #endif */ constexpr ll mod = 1e9+7; ll power(ll a, ll b){ ll res = 1; while(b){ if(b&1) res = res * a % mod; a = a * a % mod; b >>= 1; } return res; } void dzik77() { int n; cin >> n; vector<ll> a(n); ll S2 = 0; for(int i = 0; i < n; i++){ cin >> a[i]; if(a[i] > 0){ if(i==0||i==n-1) S2 +=2; else S2++; } } ll minP = 3; ll Sa = accumulate(all(a), 0LL); while(minP*(minP-1)*(minP-2)/6 < Sa) minP++; minP = max(minP, S2); ll maxP = 1000000ll; while(minP < maxP){ ll midP = (minP + maxP) / 2; ll sum = 0; ll LS = 0; bool ok = 1; for (int i = 0; i < n; i++){ if(a[i] == 0) continue; if(i == 0){ ll l = 2, r = midP; ll m = r; if((m>=3?m*(m-1)*(m-2)/6:0) + (m>=2?m*(m-1)/2*(LS):0) < a[i]){ ok = 0; break; } while(l < r){ ll m = (l + r) / 2; if((m>=3?m*(m-1)*(m-2)/6:0) + (m>=2?m*(m-1)/2*(midP - m):0) >= a[i]) r = m; else l = m + 1; } LS += l; sum += l; } else if(i == n-1){ ll l = 2, r = midP; ll m = r; if((m>=3?m*(m-1)*(m-2)/6:0) + (m>=2?m*(m-1)/2*(LS):0) < a[i]){ ok = 0; break; } while(l < r){ ll m = (l + r) / 2; if((m>=3?m*(m-1)*(m-2)/6:0) + (m>=2?m*(m-1)/2*(LS):0) >= a[i]) r = m; else l = m + 1; } LS += l; sum += l; } else{ ll l = 1, r = midP-LS; ll m = r; if((m>=3?m*(m-1)*(m-2)/6:0) + (m>=2?m*(m-1)/2*(LS):0) + m * LS * (midP - LS - m) < a[i]){ ok = 0; break; } while(l < r){ ll m = (l + r) / 2; if((m>=3?m*(m-1)*(m-2)/6:0) + (m>=2?m*(m-1)/2*(midP - m):0) + m * LS * (midP - LS - m) >= a[i]) r = m; else l = m + 1; } LS += l; sum += l; } } if(sum <= midP && ok) maxP = midP; else minP = midP+1; } cout << minP << '\n'; } int32_t main() { ios::sync_with_stdio(0); cin.tie(0); int t = 1; cin >> t; while(t--) dzik77(); 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 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 | #include <bits/stdc++.h> using namespace std; #define ll long long #define ld long double #define all(a) (a).begin(), (a).end() #define allr(a) (a).rbegin(), (a).rend() #define F first #define S second #define pb push_back typedef pair<int, int> pii; typedef pair<ll, ll> pll; /* #ifndef ONLINE_JUDGE #include <cpp-dump.hpp> #endif */ constexpr ll mod = 1e9+7; ll power(ll a, ll b){ ll res = 1; while(b){ if(b&1) res = res * a % mod; a = a * a % mod; b >>= 1; } return res; } void dzik77() { int n; cin >> n; vector<ll> a(n); ll S2 = 0; for(int i = 0; i < n; i++){ cin >> a[i]; if(a[i] > 0){ if(i==0||i==n-1) S2 +=2; else S2++; } } ll minP = 3; ll Sa = accumulate(all(a), 0LL); while(minP*(minP-1)*(minP-2)/6 < Sa) minP++; minP = max(minP, S2); ll maxP = 1000000ll; while(minP < maxP){ ll midP = (minP + maxP) / 2; ll sum = 0; ll LS = 0; bool ok = 1; for (int i = 0; i < n; i++){ if(a[i] == 0) continue; if(i == 0){ ll l = 2, r = midP; ll m = r; if((m>=3?m*(m-1)*(m-2)/6:0) + (m>=2?m*(m-1)/2*(LS):0) < a[i]){ ok = 0; break; } while(l < r){ ll m = (l + r) / 2; if((m>=3?m*(m-1)*(m-2)/6:0) + (m>=2?m*(m-1)/2*(midP - m):0) >= a[i]) r = m; else l = m + 1; } LS += l; sum += l; } else if(i == n-1){ ll l = 2, r = midP; ll m = r; if((m>=3?m*(m-1)*(m-2)/6:0) + (m>=2?m*(m-1)/2*(LS):0) < a[i]){ ok = 0; break; } while(l < r){ ll m = (l + r) / 2; if((m>=3?m*(m-1)*(m-2)/6:0) + (m>=2?m*(m-1)/2*(LS):0) >= a[i]) r = m; else l = m + 1; } LS += l; sum += l; } else{ ll l = 1, r = midP-LS; ll m = r; if((m>=3?m*(m-1)*(m-2)/6:0) + (m>=2?m*(m-1)/2*(LS):0) + m * LS * (midP - LS - m) < a[i]){ ok = 0; break; } while(l < r){ ll m = (l + r) / 2; if((m>=3?m*(m-1)*(m-2)/6:0) + (m>=2?m*(m-1)/2*(midP - m):0) + m * LS * (midP - LS - m) >= a[i]) r = m; else l = m + 1; } LS += l; sum += l; } } if(sum <= midP && ok) maxP = midP; else minP = midP+1; } cout << minP << '\n'; } int32_t main() { ios::sync_with_stdio(0); cin.tie(0); int t = 1; cin >> t; while(t--) dzik77(); return 0; } |