結果

問題 No.2493 K-th in L2 with L1
ユーザー t98slidert98slider
提出日時 2023-12-27 05:09:03
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 1,092 bytes
コンパイル時間 2,453 ms
コンパイル使用メモリ 213,748 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-09-27 15:21:10
合計ジャッジ時間 3,866 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 WA -
testcase_02 WA -
testcase_03 WA -
testcase_04 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using ll = long long;

int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    int Q;
    cin >> Q;
    auto f = [&](pair<int,int> x){
        return x.first * x.first + x.second * x.second;
    };
    while(Q--){
        int D, K;
        cin >> D >> K;
        vector<pair<int,int>> p;
        p.reserve(2 * K);
        for(int i = 0; 2 * i <= K; i++){
            p.emplace_back(i, D - i);
            p.emplace_back(i, -(D - i));
            p.emplace_back(D - i, i);
            p.emplace_back(-(D - i), i);
        }
        sort(p.begin(), p.end());
        p.erase(unique(p.begin(), p.end()), p.end());
        sort(p.begin(), p.end(), [&](pair<int,int> lhs, pair<int,int> rhs){
            return f(lhs) < f(rhs);
        });
        K--;
        if(K >= p.size()){
            cout << "No\n";
            continue;
        }
        bool ans = true;
        if(K >= 1 && f(p[K]) != f(p[K - 1])) ans = false;
        cout << (ans ? "Yes" : "No") << '\n';
        if(ans) cout << p[K].first << ' ' << p[K].second << '\n';
    }
}
0