結果

問題 No.2493 K-th in L2 with L1
コンテスト
ユーザー umimel
提出日時 2023-10-06 21:35:20
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 4 ms / 2,000 ms
コード長 1,855 bytes
コンパイル時間 1,731 ms
コンパイル使用メモリ 177,116 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-07-26 15:50:39
合計ジャッジ時間 2,226 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 4
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using pll = pair<ll, ll>;
#define drep(i, cc, n) for (ll i = (cc); i <= (n); ++i)
#define rep(i, n) drep(i, 0, n - 1)
#define all(a) (a).begin(), (a).end()
#define pb push_back
#define fi first
#define se second

mt19937_64 rng(chrono::system_clock::now().time_since_epoch().count());
const ll MOD1000000007 = 1000000007;
const ll MOD998244353 = 998244353;
const ll MOD[3] = {999727999, 1070777777, 1000000007};
const ll LINF = 1LL << 60;
const int IINF = (1 << 30) - 1;

template<typename T> struct Edge{
    int to; T w;
    Edge(int to_, T w_=1){
        to = to_;
        w=w_;
    }
};
template<typename T> using Tree = vector<vector<Edge<T>>>;
template<typename T> using Graph = vector<vector<Edge<T>>>;
/* 容量&重み付きエッジ for Dinic */
template<typename T> struct REdge{
    int to;
    T cap;
    T cost;
    int rev;
    REdge(int to_, T cap_, T cost_=1){
        to = to_;
        cap = cap_;
        cost = cost_;
    }
    
    REdge(int to_, T cap_, T cost_, int rev_){
        to = to_;
        cap = cap_;
        cost = cost_;
        rev = rev_;
    }
};

/* 残余グラフ for Dinic */
template<typename T> using RGraph = vector<vector<REdge<T>>>;

int main(){
    cin.tie(nullptr);
    ios::sync_with_stdio(false);
    
    ll Q; cin >> Q;
    while(Q--){
        ll D, K; cin >> D >> K;
        vector<tuple<ll, ll, ll>> good;
        for(ll x=-D; x<=D; x++){
            ll y = D - abs(x);
            good.pb({x*x+y*y, x, y});
            if(y>0) good.pb({x*x+y*y, x, -y});
        }
        sort(all(good));

        if(K <= (ll)good.size()){
            ll x = get<1>(good[K-1]);
            ll y = get<2>(good[K-1]);
            cout << "Yes\n";
            cout << x << " " << y << '\n';
        }else{
            cout << "No\n";
        }
    }
}
0