vector表达二维数组

#include "bits/stdc++.h"
#define int long long
using namespace std;
signed main()
{
	ios::sync_with_stdio(0);
	cin.tie(0),cout.tie(0);
	int t;
	cin>>t;
	while(t--)
	{
		int n;
		cin>>n;
		vector<vector<int>>pt(n+1,vector<int>(5));
		for(int i=1,x,y;i<=n;i++)
		{
			cin>>x>>y;
			pt[x][y]++;
		}
		int ans=0;
		for(int i=0;i<=n;i++)
		{
			int a=pt[i][0],b=pt[i][1];
			ans+=a*b*(n-a-b);
		}
		for(int i=0;i<=n-2;i++)
		{
			ans+=pt[i][0]*pt[i+2][0]*pt[i+1][1];
			ans+=pt[i][1]*pt[i+2][1]*pt[i+1][0];	
		}
		cout<<ans<<'\n';
	}
}

Problem - 2009D - Codeforces