Gym​ ​ 101908-K Kepler

题意

给你n个圆，这n个圆都包含点(0,0)。求这n个圆交点的个数。保证不存在一个点被三个圆所交。大于2n个交点输出“greater”。

思路

数据出的比较水。O(n^2)枚举再加一些剪枝是可以过掉的，但很容易被卡掉。比如出一组150000个同心圆，每两个圆的半径差值为0.001。
思路来自 black_horse2014，代码code 43676730。
所有圆都包含点(0,0)，因此可以将圆按半径排序，一层一层向外枚举。
对于每一个圆，记录其包含的圆的编号，只需要记录其下一层的圆即可。如果枚举的圆与该圆没有交点，那么也不会与其内含的圆有交点。
如果与该圆有交点，那么继续递归判断是否与它内含的圆有交点。
记录每次枚举过后，构成最外侧边界的圆的集合。这样下次枚举就先判断是否与外侧边界的圆有交点。
注意剪枝，如果当前计算出的答案大于2n的话直接退出。
代码中lev数组表示该圆是否被当前枚举的圆访问过。这样避免了重复访问。
代码：

#include <bits/stdc++.h>

using namespace std;

#define ll long long
#define ld double
#define ull unsigned long long
#define __ ios::sync_with_stdio(0);cin.tie(0);cout.tie(0)

const int maxn = 5e5 + 10;
const ll mod = 1e9 + 7;

int n;
ld X[maxn], Y[maxn], R[maxn];
int r[maxn], ans, lev[maxn], dep;
set<int> S;
vector<int> in[maxn], v;

const int cmp(const int i, const int j) {
    return R[i] < R[j];
}

bool inter(int i, int j) {
    if ((R[i] - R[j]) * (R[i] - R[j]) - (X[i] - X[j]) * (X[i] - X[j]) - (Y[i] - Y[j]) * (Y[i] - Y[j]) < 0.0)return 1;
    return 0;
}

void dfs(int j, int x) {
    lev[j] = dep;
    if (inter(j, x)) {
        ans++;
        if (ans > n)return;
        for (int i = 0; i < in[j].size(); ++i) {
            if (lev[in[j][i]] != dep) {
                dfs(in[j][i], x);
            }
        }
    } else in[x].push_back(j);
}

int main() {
    __;
    cin >> n;
    for (int i = 1; i <= n; ++i) {
        cin >> X[i] >> Y[i] >> R[i];
        r[i] = i;
    }
    sort(r + 1, r + 1 + n, cmp);
    for (int i = 1; i <= n; ++i) {
        int x = r[i];
        if (v.size())v.clear();
        for (auto j:S) {
            if (!inter(j, x))v.push_back(j);
            ++dep;
            dfs(j, x);
        }
        for (int j:v)S.erase(j);
        if (ans > n)break;
        S.insert(x);
    }
    if (ans > n)cout << "greater" << endl;
    else cout << ans * 2 << endl;
    return 0;
}