Educational Codeforces Round 115 (Rated for Div. 2) D. Training Session

题目链接:D. Training Session
题目大意: 给你n个(x,y)对,其中x是题目的主题 y的题目的难度 ,从中找出三个(x,y)对,要求这三个对要么是x都不同,要么是y都不同,要么是x和y都不相同。

思路: 很容易知道总的选择数就是 C n 3 C_{n}^{3} Cn3​也就是 n ∗ ( n − 2 ) ∗ ( n − 2 ) 6 {n*(n-2)*(n-2)\over 6} 6n∗(n−2)∗(n−2)​,那么只要找出不符合的然后相减就能得出最终答案了。
那么怎么找出不符合的呢,根据题目不可能存在完全一致的(x,y)对,所以我们可以知道不符合题意的一定是这种形式的 [ X Y X a n t h i n g a n t h i n g Y ] \begin{bmatrix}X & Y \\ X & anthing\\ anthing & Y\end{bmatrix} ⎣⎡​XXanthing​YanthingY​⎦⎤​

现在说一下为什么一定是这种形式的。
根据题意左边至少有两个数相同,右边也至少有两个数相同。那么当为左边三个数全部相同的时候呢?答案是这种情况只有可能是符合题意的。因为当最左边存在三个数相同时右边同时也存在两个数相同,那么就必定有两个对是一模一样的,这和题目描述不同。那么之后就很容易算了,我们只需要将每一组(x,y)中x和y的出现次数记录下来就可以了。举个例子:
[ X Y X 1 1 Y X 2 X 3 2 Y 3 Y ] \begin{bmatrix}X & Y \\ X & 1\\ 1 & Y\\X&2\\X&3\\2&Y\\3&Y\end{bmatrix} ⎣⎢⎢⎢⎢⎢⎢⎢⎢⎡​XX1XX23​Y1Y23YY​⎦⎥⎥⎥⎥⎥⎥⎥⎥⎤​
现在找不符合题意的:
第一组是(X,Y)
那么第二组可以选(X,1)、(X,2)、(X,3)
第三组可以选(1,Y)、(2,Y)、(3,Y)
所以当第一组是(X,Y)时不符合题意的种类数就是(cnt[x]-1)*(cnt[y]-1)那么之后只要遍历一遍就可以找出所有的不符合题意的组数。

代码:

#include <bits/stdc++.h>
using namespace std;
const int N=2e5+10;
typedef long long ll;
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    int t;
    cin>>t;
    while(t--)
    {
        int n;
        cin>>n;
        vector<int> a(n+10),b(n+10),cl(n+10),cr(n+10);//用普通数组的话需要memset,之后就超时了。
        for(int i=1;i<=n;i++)
        {
            cin>>a[i]>>b[i];
            cl[a[i]]++,cr[b[i]]++;//记住左右两边值出现的次数
        }
        ll sum=(ll)n*(n-1)*(n-2)/6;//总数,记得转long long 
        for(int i=1;i<=n;i++)
        {
            sum-=(ll)(cl[a[i]]-1)*(cr[b[i]]-1);//每一对能够组成不符合题意的种类数
        }
        cout<<sum<<endl;
    }
}

上一篇:剑指 Offer 56 - II. 数组中数字出现的次数 II


下一篇:python 操作.mat文件