#include <bits/stdc++.h> using namespace std; int main() { ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); int tesc; cin >> tesc; vector<int> odp; while(tesc--) { int budy; cin >> budy; vector<long long> gry(budy); for (int ind = 0; ind < budy; ind++) { cin >> gry[ind]; } long long sumG = 0; int ile = 0; for (int ind = 0; ind < budy; ind++) { sumG += gry[ind]; if(gry[ind] > 0) ile++; } int dol = 3; dol = max(dol, ile); int tmp = dol; while((long long)tmp * (tmp - 1) * (tmp - 2) / 6 < sumG) tmp++; dol = tmp; int gor = dol; auto ok = [&](int tgr) -> bool { long long sump = 0; for (int ind = 0; ind < budy; ind++) { long long req = gry[ind]; bool mid = (ind != 0 && ind != budy - 1); int minx = 0, maxx = tgr, ans = tgr + 1; while(minx <= maxx) { int miv = (minx + maxx) / 2; long long wyn = 0; if(miv < 2) { if(mid) { if(miv == 1) { int dod = tgr - miv; int lew = dod / 2; int pra = dod - lew; wyn = 1LL * miv * lew * pra; } else wyn = 0; } else wyn = 0; } else { long long c2 = 1LL * miv * (miv - 1) / 2; long long c3 = (miv >= 3) ? 1LL * miv * (miv - 1) * (miv - 2) / 6 : 0; long long part = c3 + c2 * (tgr - miv); wyn = part; if(mid) { int dod = tgr - miv; int lew = dod / 2; int pra = dod - lew; wyn += 1LL * miv * lew * pra; } } if(wyn >= req) { ans = miv; maxx = miv - 1; } else { minx = miv + 1; } } sump += ans; if(sump > tgr) return false; } return sump <= tgr; }; while(!ok(gor)) gor *= 2; int res = gor; while(dol <= gor) { int sro = (dol + gor) / 2; if(ok(sro)) { res = sro; gor = sro - 1; } else { dol = sro + 1; } } odp.push_back(res); } for(auto val : odp) cout << val << endl; 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 | #include <bits/stdc++.h> using namespace std; int main() { ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); int tesc; cin >> tesc; vector<int> odp; while(tesc--) { int budy; cin >> budy; vector<long long> gry(budy); for (int ind = 0; ind < budy; ind++) { cin >> gry[ind]; } long long sumG = 0; int ile = 0; for (int ind = 0; ind < budy; ind++) { sumG += gry[ind]; if(gry[ind] > 0) ile++; } int dol = 3; dol = max(dol, ile); int tmp = dol; while((long long)tmp * (tmp - 1) * (tmp - 2) / 6 < sumG) tmp++; dol = tmp; int gor = dol; auto ok = [&](int tgr) -> bool { long long sump = 0; for (int ind = 0; ind < budy; ind++) { long long req = gry[ind]; bool mid = (ind != 0 && ind != budy - 1); int minx = 0, maxx = tgr, ans = tgr + 1; while(minx <= maxx) { int miv = (minx + maxx) / 2; long long wyn = 0; if(miv < 2) { if(mid) { if(miv == 1) { int dod = tgr - miv; int lew = dod / 2; int pra = dod - lew; wyn = 1LL * miv * lew * pra; } else wyn = 0; } else wyn = 0; } else { long long c2 = 1LL * miv * (miv - 1) / 2; long long c3 = (miv >= 3) ? 1LL * miv * (miv - 1) * (miv - 2) / 6 : 0; long long part = c3 + c2 * (tgr - miv); wyn = part; if(mid) { int dod = tgr - miv; int lew = dod / 2; int pra = dod - lew; wyn += 1LL * miv * lew * pra; } } if(wyn >= req) { ans = miv; maxx = miv - 1; } else { minx = miv + 1; } } sump += ans; if(sump > tgr) return false; } return sump <= tgr; }; while(!ok(gor)) gor *= 2; int res = gor; while(dol <= gor) { int sro = (dol + gor) / 2; if(ok(sro)) { res = sro; gor = sro - 1; } else { dol = sro + 1; } } odp.push_back(res); } for(auto val : odp) cout << val << endl; return 0; } |