hihocoder 1497 Queen Attack

排序。

分别按照$x$，$y$以及对角线排序，统计一下方案数就可以了。

#include<bits/stdc++.h>
using namespace std;

int n;
struct X
{
    int x,y;
}s[];

bool cmp1(X a, X b)
{
    return a.x<b.x;
}

bool cmp2(X a, X b)
{
    return a.y<b.y;
}

bool cmp3(X a, X b)
{
    return a.x+a.y<b.x+b.y;
}

bool cmp4(X a, X b)
{
    return a.y-a.x<b.y-b.x;
}

int main()
{
    scanf("%d",&n);
    for(int i=;i<=n;i++) scanf("%d%d",&s[i].x,&s[i].y);

    long long sum=;

    sort(s+,s++n,cmp1);
    long long a=;
    for(int i=;i<=n;i++)
    {
        if(s[i].x==s[i-].x) a++;
        else
        {
            sum=sum+a*(a-)/;
            a=;
        }
    }
    sum=sum+a*(a-)/;

    sort(s+,s++n,cmp2);
    a=;
    for(int i=;i<=n;i++)
    {
        if(s[i].y==s[i-].y) a++;
        else
        {
            sum=sum+a*(a-)/;
            a=;
        }
    }
    sum=sum+a*(a-)/;

    sort(s+,s++n,cmp3);
    a=;
    for(int i=;i<=n;i++)
    {
        if(s[i].y+s[i].x==s[i-].y+s[i-].x) a++;
        else
        {
            sum=sum+a*(a-)/;
            a=;
        }
    }
    sum=sum+a*(a-)/;

    sort(s+,s++n,cmp4);
    a=;
    for(int i=;i<=n;i++)
    {
        if(s[i].y-s[i].x==s[i-].y-s[i-].x) a++;
        else
        {
            sum=sum+a*(a-)/;
            a=;
        }
    }
    sum=sum+a*(a-)/;

    printf("%lld\n",sum);

    return ;
}