Codeforces 862div2 C
// compile: g++ -o data data.cpp -std=gnu++20 -DLOCAL
// run: ./data < data.in
#include <bits/stdc++.h>
#pragma GCC optimize("O3,unroll-loops")
#pragma GCC target("avx2,bmi,bmi2,lzcnt,popcnt")
#ifdef LOCAL
#endif
#define sz(v) ((int)(v).size())
#define PV(v) for (int i=0;i<v.size();++i)cout<<v[i]<<" ";cout<<endl;
#define INTINF 9223372036854775807
#define int long long

using namespace std;

int n, m;
struct parabola {
    int a, b, c;
};
vector<int> l;
vector<parabola> p;

bool intersect(int k, parabola p) {
    double x = (k - p.b)*1.0 / (2 * p.a);
    double minv = p.a*x*x + (p.b-k)*x + p.c;
    return minv <= 0;
}

int gt(double k, vector<int> &a) {
    if (!sz(a)) return -1;
    int l = 0, r = sz(a) - 1, ans = -1;
    while (l <= r) {
        int mid = l + (r - l) / 2;
        if (a[mid] <= k) l = mid + 1;
        else ans = mid, r = mid - 1;
    }
    return ans;
}

int lt(double k, vector<int> &a) {
    if (!sz(a)) return -1;
    int l = 0, r = sz(a) - 1, ans = -1;
    while (l <= r) {
        int mid = l + (r - l) / 2;
        if (a[mid] >= k) r = mid - 1;
        else ans = mid, l = mid + 1;
    }
    return ans;
}

pair<double, double> tang(parabola p) {
    double x1 = sqrt(fabs(p.c*1.0 / p.a));
    double x2 = -sqrt(fabs(p.c*1.0 / p.a));
    double k1 = 2*p.a*x1 + p.b;
    double k2 = 2*p.a*x2 + p.b;
    return make_pair(k1, k2);
}

int32_t main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    cout.tie(nullptr);
    int _; cin >> _;
    while (_--) {
        // pos.clear(); neg.clear(); p.clear();
        l.clear(); p.clear();
        cin >> n >> m;
        for (int i = 0; i < n; ++i) {
            int k; cin >> k;
            l.push_back(k);
        }
        sort(l.begin(), l.end());
        for (int i = 0; i < m; ++i) {
            int a, b, c; cin >> a >> b >> c;
            p.push_back(parabola{a, b, c});
            pair<double, double> k = tang(parabola{a, b, c});
            int x1 = gt(k.first, l);
            int y1 = lt(k.first, l);
            int x2 = gt(k.second, l);
            int y2 = lt(k.second, l);
            if (x1 != -1 && !intersect(l[x1], parabola{a, b, c})) cout << "YES\n" << l[x1] << endl;
            else if (x2 != -1 && !intersect(l[x2], parabola{a, b, c})) cout << "YES\n" << l[x2] << endl;
            else if (y1 != -1 && !intersect(l[y1], parabola{a, b, c})) cout << "YES\n" << l[y1] << endl;
            else if (y2 != -1 && !intersect(l[y2], parabola{a, b, c})) cout << "YES\n" << l[y2] << endl;
            else cout << "NO" << endl;
           
            // int x = gt(t, l);
            // int y = lt(t, l);
            // if (x != -1 && !intersect(l[x], parabola{a, b, c})) cout << "YES\n" << l[x] << endl;
            // else if (y != -1 && !intersect(l[y], parabola{a, b, c})) cout << "YES\n" << l[y] << endl;
            // else cout << "NO" << endl;
        }
    }

    return 0;
}
No Comments

Send Comment Edit Comment


				
|´・ω・)ノ
ヾ(≧∇≦*)ゝ
(☆ω☆)
(╯‵□′)╯︵┴─┴
 ̄﹃ ̄
(/ω\)
∠( ᐛ 」∠)_
(๑•̀ㅁ•́ฅ)
→_→
୧(๑•̀⌄•́๑)૭
٩(ˊᗜˋ*)و
(ノ°ο°)ノ
(´இ皿இ`)
⌇●﹏●⌇
(ฅ´ω`ฅ)
(╯°A°)╯︵○○○
φ( ̄∇ ̄o)
ヾ(´・ ・`。)ノ"
( ง ᵒ̌皿ᵒ̌)ง⁼³₌₃
(ó﹏ò。)
Σ(っ °Д °;)っ
( ,,´・ω・)ノ"(´っω・`。)
╮(╯▽╰)╭
o(*////▽////*)q
>﹏<
( ๑´•ω•) "(ㆆᴗㆆ)
😂
😀
😅
😊
🙂
🙃
😌
😍
😘
😜
😝
😏
😒
🙄
😳
😡
😔
😫
😱
😭
💩
👻
🙌
🖕
👍
👫
👬
👭
🌚
🌝
🙈
💊
😶
🙏
🍦
🍉
😣
Source: github.com/k4yt3x/flowerhd
颜文字
Emoji
小恐龙
花!
Previous
Next