公式用的容斥原理的变式,求大佬帮调

#include<bits/stdc++.h>
using namespace std;
long long r[1000005],c[1000005];
long long ans;
int main()
{
	long long n,k;
	scanf("%lld%lld",&n,&k);
	for(int i=1;i<=k;i++)
		scanf("%lld%d",r+i,c+i);
	sort(r+1,r+1+k);
	sort(c+1,c+1+k);
	int n1=unique(r+1,r+1+k)-(r+1);
	int n2=unique(c+1,c+1+k)-(c+1);
	ans=n*n1+n*n2-n1*n2;
	printf("%lld",ans);
	return 0;
}
/*
N×N 的国际象棋棋盘上有K 个车
3 2
1 2
2 2
*/