結果

問題 No.461 三角形はいくつ？
ユーザー pekempey
提出日時 2016-12-12 01:53:47
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 46 ms / 5,000 ms
コード長 1,863 bytes
コンパイル時間 2,931 ms
コンパイル使用メモリ 188,580 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-11-29 12:08:56
合計ジャッジ時間 5,297 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 41
権限があれば一括ダウンロードができます

ソースコード

diff # 

#pragma GCC optimize ("O3")
#pragma GCC target ("avx")
#include <bits/stdc++.h>
using namespace std;

struct Fraction {
    int64_t a, b;

    Fraction(int64_t a, int64_t b) : a(a), b(b) {}
};

bool operator<(const Fraction &l, const Fraction &r) {
    return l.a * r.b < l.b * r.a;
}

bool operator>=(const Fraction &l, const Fraction &r) {
    return l.a * r.b >= l.b * r.a;
}

Fraction operator+(const Fraction &l, const Fraction &r) {
    return Fraction(l.a * r.b + r.a * l.b, l.b * r.b);
}

Fraction operator-(const Fraction &l, const Fraction &r) {
    return Fraction(l.a * r.b - r.a * l.b, l.b * r.b);
}

bool operator==(const Fraction &l, const Fraction &r) {
    return l.a * r.b == r.a * l.b;
}

int main() {
    int n;
    cin >> n;
    vector<vector<Fraction>> f(3);
    for (int i = 0; i < 3; i++) {
        f[i].emplace_back(1, 1);
    }
    for (int i = 0; i < n; i++) {
        int p, ta, tb;
        cin >> p >> ta >> tb;
        f[p].emplace_back(ta, ta + tb);
    }
    sort(f[1].begin(), f[1].end());
    sort(f[2].begin(), f[2].end());
    int64_t ans = 0;
    for (int i = 0; i < f[0].size(); i++) {
        int k0 = lower_bound(f[2].begin(), f[2].end(), Fraction(1, 1) - f[0][i]) - f[2].begin();
        int k1 = f[2].size();
        int k2 = f[2].size();
        for (int j = 0; j < f[1].size(); j++) {
            while (k1 > 0 && f[1][j] + f[2][k1 - 1] >= Fraction(1, 1)) {
                k1--;
            }
            while (k2 > 0 && f[0][i] + f[1][j] + f[2][k2 - 1] >= Fraction(2, 1)) {
                k2--;
            }
            if (f[0][i] + f[1][j] < Fraction(1, 1)) {
                continue;
            }
            if (k2 < f[2].size() && f[0][i] + f[1][j] + f[2][k2] == Fraction(2, 1)) {
                ans--;
            }
            ans += f[2].size() - max(k0, k1);
        }
    }
    cout << ans << endl;
}
0