D. Training Session(1598D)(思维、组合数学)

D. Training Session(1598D)(思维、组合数学)

---

题目来源：D. Training Session

---

题意：

给出 n 个问题，每个问题有两个属性：序号 ai 和难度 bi，其中 1 <= ai, bi <= n，保证没有两个问题的序号和难度都相同

求选出三个问题满足下面要求之一的方案数

• 三个问题的 a 全不相同
• 三个问题的 b 全不相同

---

思路：

逆向思考，用所有方案数减去不合法的方案数

那么所有方案数就是C(n, 3)

接下来考虑不合法的情况：

首先开两个数组：coua用于记录 a[i] 出现的次数，coub用于记录 b[i] 出现的次数

现在假设有三个问题x, y, z，其中 x.a = y.a, x.b = z.b（题目保证了没有两个问题是完全相同的），每次都选问题 x，不合法的数量就是去找 x.a 出现的次数 coua[x.a] 和 x.b 出现的次数 coub[x.b]，因为没有相同的问题，最后两两组合一定可以选出三个问题，所以选择 x 的不合法答案就是(coua[x.a] - 1) * (coub[x.b] - 1)

---

AC代码

#include <bits/stdc++.h>
#define endl "\n"
#define rep(i, m, n) for (int i = (m); i <= (n); ++i)
#define rrep(i, m, n) for (int i = (m); i >= (n); --i)
#define IOS ios::sync_with_stdio(0); cin.tie(0);
using namespace std;
typedef long long ll;
typedef pair<int, int> PII;
const int N = 2e5 + 10, mod = 1e9 + 7;
int n, x[N], y[N];
int coua[N], coub[N];
void solve() {
	scanf("%d", &n);
	rep(i, 1, n) coua[i] = coub[i] = 0;
	rep(i, 1, n) {
		scanf("%d %d", &x[i], &y[i]);
		coua[x[i]]++; coub[y[i]]++;
	}

	ll ans = 1ll * n * (n - 1) * (n - 2) / 6;
	rep(i, 1, n) {
		ll a = coua[x[i]];
		ll b = coub[y[i]];
		ans -= (a - 1) * (b - 1);
	}

	printf("%lld\n", ans);
}
int main() {
	int t; cin >> t;
	while (t--) solve();
	return 0;
}

---

END