Codeforces Round #544 (Div. 3):D - Zero Quantity Maximization

题目链接：http://codeforces.com/contest/1133/problem/D

题意：现在有两个数列a和b，要你找到一个实数d，a[i]*d+b[i] == 0的个数最多，问最多的个树是多少。

解题心得：

• 要想得比较仔细才不会漏掉，有以下几种情况：
  • a[i]为0，b[i]不为0，答案不可能为0
  • b[i]为0，那么不论a[i]是多少只要d为零，那么答案肯定为0
  • a[i]，b[i]同时为0，那么无论d是什么，肯定为0
• 这样讨论的时候就找max( pair(a[i], b[i])相等的最大数目，d为0且b[i]为0的时候的个数)，加上a[i]，b[i]同时为0的个数就是答案了。

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;
const int maxn = 2e5 + 100;

map<pair<int, int>, int> maps;
int a[maxn], b[maxn], n;

int abs(int x) {
    if (x < 0) return -x;
    return x;
}

void change(int&x, int &y) {//当a/b，规定当两者异号的时候a为负，当两者同号的时候同正
    if ((x > 0 && y < 0) || (x < 0 && y < 0)) {
        x = -x;
        y = -y;
    }
}

int get_ans1() {//找到一个a*d + b == 0
    int ans = 0;
    map<pair<int, int>, int>::iterator iter;
    for (iter = maps.begin(); iter != maps.end(); iter++) {
        if (iter->second > ans) {
            ans = iter->second;
        }
    }

    return ans;
}

int get_ans2() {//当b为0的时候，设d为0，又多少个
    int ans = 0;
    map<pair<int, int>, int>::iterator iter;
    for (iter = maps.begin(); iter != maps.end(); iter++) {
        if (iter->first.second == 0) ans += iter->second;
    }
    return ans;
}

int main() {
//    freopen("1.in", "r", stdin);
    scanf("%d", &n);
    for (int i = 1; i <= n; i++) {
        scanf("%d", &a[i]);
    }
    for (int i = 1; i <= n; i++) {
        scanf("%d", &b[i]);
    }

    int cnt = 0;
    for (int i = 1; i <= n; i++) {
        //a[i]*d + b[i] 不可能为0
        if (a[i] == 0 && b[i] != 0) continue;

        //a[i]*d + b[i] 一定为0
        else if (a[i] == 0 && b[i] == 0) {
            cnt++;
            continue;
        }

        //最简分数a/b
        ll gcd = __gcd(abs(a[i]), abs(b[i]));
        a[i] /= gcd;
        b[i] /= gcd;

        change(a[i], b[i]);

        maps[make_pair(a[i], b[i])]++;
    }

    int ans = get_ans1();
    int ans2 = get_ans2() + cnt;
    printf("%d", max(ans2, ans + cnt));
    return 0;
}