Codeforces 650A Watchmen 【数学 + 二分】

最新推荐文章于 2018-09-27 19:42:00 发布

笑着走完自己的路

最新推荐文章于 2018-09-27 19:42:00 发布

阅读量576

点赞数

分类专栏： codeforces 二分数学

本文链接：https://blog.csdn.net/chenzhenyu123456/article/details/50835009

版权

codeforces 同时被 3 个专栏收录

215 篇文章 1 订阅

订阅专栏

数学

94 篇文章 0 订阅

订阅专栏

二分

62 篇文章 0 订阅

订阅专栏

题目链接：Codeforces 650A Watchmen

题意：给 $n$ 个点，问满足 $|xi - xj| + |yi -yj| = \sqrt{(xi-xj)^2 + (yi-yj)^2}$ 的 $(i, j)$ 有多少对。

思路：显然当 $x$ 或者 $y$ 相等时才成立，那么先按 $x$ sort，求出 $x$ 相等情况的个数，再求出 $y$ ，最后去掉相等的点。

$AC$ 代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <vector>
#include <queue>
#include <map>
#define CLR(a, b) memset(a, (b), sizeof(a))
#define fi first
#define se second
using namespace std;
typedef long long LL;
typedef pair<int, int> pii;
const int MAXN = 2*1e5+10;
const int INF = 0x3f3f3f3f;
void getmax(int &a, int b) {a = max(a, b); }
void getmin(int &a, int b) {a = min(a, b); }
map<pii, int> fp;
pii a[MAXN], b[MAXN];
bool cmp(pii a, pii b) {
    return a.fi != b.fi ? a.fi < b.fi : a.se < b.se;
}
int main()
{
    int n; scanf("%d", &n); fp.clear();
    for(int i = 0; i < n; i++)
    {
        scanf("%d%d", &a[i].fi, &a[i].se);
        fp[a[i]]++;
        b[i].fi = a[i].se; b[i].se = a[i].fi;
    }
    sort(a, a+n, cmp);
    LL ans = 0;
    for(int i = 0; i < n; i++)
    {
        int pos = lower_bound(a+i, a+n, pii(a[i].fi+1, -INF)) - a;
        //cout << pos << endl;
        if(pos > i) ans += 1LL * (pos - i - 1);
    }
    //cout << ans << endl;
    sort(b, b+n, cmp);
    for(int i = 0; i < n; i++)
    {
        int pos = lower_bound(b+i, b+n, pii(b[i].fi+1, -INF)) - b;
        if(pos > i) ans += 1LL * (pos - i - 1);
    }
    //cout << ans << endl;
    int top = unique(a, a+n) - a;
    for(int i = 0; i < top; i++)
        ans -= 1LL * (fp[a[i]] - 1) * fp[a[i]] / 2;
    cout << ans << endl;
    return 0;
}