Codeforces Round 971 (Div. 4) D. Satyam and Counting

打了场div4 来补下题
D. Satyam and Counting
题目：
萨提亚姆在二维坐标平面上给出了 n 个不同的点。对于所有给定点 (xi,yi)，保证 0≤yi≤1,选择三个不同的点作为顶点，可以形成多少个不同的非世代直角三角形？如果有一个点 v 使得 v 是 a的顶点而不是 b的顶点，则两个三角形 a和 b 是不同的。 一个非整角直角三角形的面积为正，内角为 90∘

输入

第一行包含一个整数 t( 1≤t≤104 ) - 测试用例的数量。
每个测试用例的第一行包含一个整数 n ( 3≤n≤2⋅105) 点数。
接下来的 n 行包含两个整数 xi和 yi ( 0≤xi≤n , 0≤yi≤1)--Satyam 可以选择的 i'th 点。保证所有 (xi,yi) 都是成对的不同点。
保证所有测试用例中 n 的总和不超过 2⋅105。

输出
为每个测试用例输出一个整数，即从选择的三个点中可以形成的不同非分解直角三角形的个数。

样例：
Input
3
5
1 0
1 1
3 0
5 0
2 1
3
0 0
1 0
3 0
9
1 0
2 0
3 0
4 0
5 0
2 1
7 1
8 1
9 1
Output
4
0
8

思路：刚开始看到这个题无从下手,我们先读一遍题,题目的突破点是0≤yi≤1，y轴的坐标只能是0或者1，那么好，我们第一种想到的情况是两个点的x坐标相同，y的坐标一个是0一个是1，那么这两个点和其他任何的点构成的三角形都是直角三角形,我们就找这种点，找到一个答案就加上n-2
第二种情况是
2，√2, √2的情况,
只要有三个点的x在范围内,并且其中两个点的y轴和另外的一个点的y轴不同即可，找到一个，答案就加上1

#include<bits/stdc++.h>
using namespace std;
 
typedef long long LL;
typedef pair<int,int> PII;
typedef pair<long long,long long> PLL;
typedef unordered_map<int,int> HashII;
typedef unordered_map<char,int> HashCI;
typedef priority_queue<int,vector<int>,greater<int>> heap;
typedef unsigned long long ULL;
typedef unordered_map<string,int> HashSI;
const int INF=0x3f3f3f3f;
const int N=2e5+10;
#define x first
#define y second
void solve()
{
    //---Dijkstra
    LL ans = 0;
    int n; cin>>n;
    vector<PLL> a(n+1);
    vector<int> c0(n+1),c1(n+1);//这个我们用来统计当坐标x为i的时候，这个点的y坐标是0还是1
    LL MMax = -INF;
    for(int i=0;i<n;i++){
        cin>>a[i].x>>a[i].y;
        if(a[i].y==0) c0[a[i].x]++;
        if(a[i].y==1) c1[a[i].x]++;
        MMax = max(MMax,a[i].x);
    }
    for(int i=0;i<=MMax;i++)
    {
        if(c0[i] && c1[i]) ans+=n-2;//找第一种情况
        if(i-2>=0 && c0[i] && c0[i-2] && c1[i-1]) ans++;//第二种情况
        if(i-2>=0 && c1[i] && c1[i-2] && c0[i-1]) ans++;
    }
    cout<<ans<<"\n";
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(nullptr),cout.tie(nullptr);
    int t ;
    cin>>t;
    while(t--)
    {
        solve();
    }
    return 0;
}