Submission #402371

---

Source Code Expand

#include <iostream>
#include <stdio.h>
#include <string.h>
using namespace std;

	int r[110000];
	int c[110000];
	int rs[110000];
	int cs[110000];
	int rc[110000];
	int cc[110000];
int main(void)
{
	int R, C, K, N;

	cin >> R >> C >> K >> N;
	int i;
	for(i = 0 ; i < N ; i++)
	{
		cin >> r[i] >> c[i];
		r[i]--;c[i]--;
	}
	for(i = 0 ; i < R ; i++)
	{
		rs[i] = 0;
	}
	for(i = 0 ; i < C ; i++)
	{
		cs[i] = 0;
	}
	for(i = 0 ; i < N ; i++)
	{
		rs[r[i]]++;
		cs[c[i]]++;
	}
	long long sum = 0;
	for(i = 0 ; i < N ; i++)
	{
		if(rs[r[i]] + cs[c[i]] == K+1)
		{
			sum++;
		}
		if(rs[r[i]] + cs[c[i]] == K)
		{
			sum--;
		}
	}
	for(i = 0 ; i < 100001 ; i++)
	{
		rc[i] = 0;
		cc[i] = 0;
	}
	for(i = 0 ; i < R ; i++)
	{
		rc[rs[i]]++; 
	}
	for(i = 0 ; i < C ; i++)
	{
		cc[cs[i]]++; 
	}
	for(i = 0 ; i <= K ; i++)
	{
		sum += (long long)rc[i] * cc[K-i];
	}
	cout << sum << endl;
	return 0;
}

Submission Info

Judge Result