// モジュール // #REQ: base_template.cpp // // using ll = long signed long; using ull = long unsigned long; using ld = long double; // #include using namespace std; // // #REQ: base_template/575 base_template/ll // 汎用マクロ #define ALL(x) (begin(x)), (end(x)) #define ALLOF(x) (begin(x)), (end(x)) #define RALL(x) (rbegin(x)), (rend(x)) #define RALLOF(x) (rbegin(x)), (rend(x)) #define RPE REP #define ERP REP #define EPR REP #define PER REP #define PRE REP #define REP(i,n) for (ll i=0, i##_len=(n); i=i##_end; i--) #define STEP(i, is, ie, step) for (ll i=(is), i##_end=(ie), i##_step = (step); i<=i##_end; i+=i##_step) #define UNIQUE(v) do { sort((v).begin(), (v).end()); (v).erase(unique((v).begin(), (v).end()), (v).end()); } while (false) #define FOREACH(i,q) for (auto &i : q) template bool chmax(T &a, const U b) { if (a < b) {a = b; return true;} return false; } template bool chmin(T &a, const U b) { if (a > b) {a = b; return true;} return false; } constexpr int INF = numeric_limits::max(); constexpr ll LINF = numeric_limits::max(); #define Yes(q) ((q) ? "Yes" : "No") #define YES(q) ((q) ? "YES" : "NO") #define Possible(q) ((q) ? "Possible" : "Impossible") #define POSSIBLE(q) ((q) ? "POSSIBLE" : "IMPOSSIBLE") #define IIF(q,t,f) ((q) ? (t) : (f)) // // AtCoder/Codeforces 用 デバッグ検知 #ifdef ONLINE_JUDGE constexpr bool DEBUG_MODE = false; #else constexpr bool DEBUG_MODE = true; #endif // // #REQ: base_template/575 base_template/ll base_template/macro // 標準入出力 struct qin { // query input size_t sz; qin(size_t _sz = 1) : sz(_sz) {} template operator T () const { T a; cin >> a; return a; } template operator vector () const { vector a(sz); for (size_t i = 0; i < sz; i++) cin >> a[i]; return a; } template operator pair () const { T f; U s; cin >> f >> s; return pair(f, s); } }; qin in1; // input one template void say(const T x, const char* end = "\n") { cout << x << end; } void say(const ld x, const char* end = "\n") { cout << setprecision(30) << x << end; } template void say(const vector x, const char* sep = " ", const char* end = "\n") { REP(i, x.size()) { cout << x[i] << (i+1 == i_len ? end : sep); } } template void say(const vector> x, const char* sep = " ", const char* end = "\n") { REP(i, x.size()) { say(x[i], sep, end); } } // // #REQ: base_template/575 base_template/debug_mode #define DUMP(q) DUMP_FUNC(q, #q, __FILE__, __LINE__) template void DUMP_PROC(T x) { if (is_integral() || is_floating_point()) cerr << "\e[32m" << x << "\e[m"; else cerr << x; } template<> void DUMP_PROC(char x) { cerr << "\e[36m\'" << x << "\'\e[m"; } template<> void DUMP_PROC(string x) { cerr << "\e[33m\"" << x << "\"\e[m"; } template void DUMP_PROC(pair x) { cerr << "{"; DUMP_PROC(x.first); cerr << ", "; DUMP_PROC(x.second); cerr << "}"; } template void DUMP_PROC(tuple &x, integer_sequence) { (void)(int[]){(cerr << ((const char*[]){"", ", "})[!!Seq] << (DUMP_PROC(get(x)), ""), 0)...}; } template void DUMP_PROC(tuple x) {cerr << "{"; DUMP_PROC(x, index_sequence_for()); cerr << "}";} template void DUMP_PROC(vector x) { cerr << "["; for (auto &xi : x) { DUMP_PROC(xi); cerr << (&xi != &*x.rbegin()?", ":""); } cerr << "]"; } template void DUMP_FUNC(T x, const char* name, const char* fn, int ln) { if (!DEBUG_MODE) return; cerr << "\e[32m[DEBUG]\e[m " << name << ": "; DUMP_PROC(x); cerr << " @ " << fn << "(" << ln << ")" << endl; } #define DUMPT(...) DUMP(tuple(__VA_ARGS__)) // // gcc拡張マクロ #define popcount __builtin_popcount #define popcountll __builtin_popcountll // // #REQ: base_template/575 base_template/ll base_template/debug_mode base_template/gcc base_template/macro base_template/dump base_template/qin // base template set // // #END void solve() { ll x, a; cin >> x >> a; ll w = ceil(sqrt(x)) + 10; vector dp(a+1, LINF); dp[0] = 0; REP(i, a+1) { RANGE(k, 1, min(i, w)) { if (dp[i-k] < LINF) { chmin(dp[i], dp[i-k] + k * k + x); } } } RANGE(k, 1, a) { if (dp[a-k] < LINF) { chmin(dp[a], dp[a-k] + k * k + x); } } say(dp[a]); } // 処理内容 int main() { ios::sync_with_stdio(false); // stdioを使うときはコメントアウトすること cin.tie(nullptr); // インタラクティブ問題ではコメントアウトすること // code here ll t = in1; RPE(i, t) solve(); }