結果

問題 No.2493 K-th in L2 with L1
ユーザー t98slider
提出日時 2023-12-27 05:14:23
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 964 bytes
コンパイル時間 2,094 ms
コンパイル使用メモリ 204,816 KB
最終ジャッジ日時 2025-02-18 14:46:15
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 1
other AC * 1 WA * 3
権限があれば一括ダウンロードができます

ソースコード

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(4 * D);
        for(int i = 0; i <= D; i++){
            p.emplace_back(i, D - i);
            p.emplace_back(-i, -(D - i));
            p.emplace_back(i, D - i);
            p.emplace_back(-i, -(D - 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;
        }
        cout << "Yes\n" << p[K].first << ' ' << p[K].second << '\n';
    }
}
0