EOJ 3346. 皇后问题 （标记法）

#include <bits/stdc++.h>

using namespace std;
#define N 100010
typedef long long ll;

ll row[N];//行
ll col[N];//列
ll dia1[2*N];//副对角线 y = -x + b
ll dia2[2*N];//主对角线 y =  x + b

int main()
{
    int n;
    cin >> n;
    if(n == 1)
        cout << 0 << endl;
    else
    {
        int x, y;
        for(int i = 0; i < n; i++)
        {
            cin >> x >> y;
            row[x]++;
            col[y]++;
            dia1[x+y]++;
            dia2[y-x+N]++;//+N用于偏移 防止数组下标为负
        }
        ll ans = 0;
        for(int i = 0; i < N; i++)
        {
            if(row[i] >= 2)
                ans += row[i] * (row[i] - 1) / 2;//计算每行冲突数
            if(col[i] >= 2)
                ans += col[i] * (col[i] - 1) / 2;//计算每列冲突数
        }
        for(int i = 1; i < 2*N; i++)
        {
            if(dia1[i] >= 2)
                ans += dia1[i] * (dia1[i] - 1) / 2;//计算每副对角线冲突数
            if(dia2[i] >= 2)
                ans += dia2[i] * (dia2[i] - 1) / 2;//计算每主对角线冲突数
        }
        cout << ans << endl;
    }

    return 0;
}