2021-01-25 cf #697 Div3 C题(超时,换思路减少复杂度)

题目链接:https://codeforces.com/contest/1475/problem/C

题意要求:需组成的2对,男的序号不能重,女的序号不能重

比如这例

输入:	行1--测试个数 　　行1` --男生个数p,女生个数q,成对数k 　　接下两行`--分别为这k对男女序号(上下为一对)	1 3 4 4 1 1 2 3 2 3 2 4
输出:	合法的组队对数	4

• (1,2) and (3,4);
• (1,3) and (2,2);
• (1,3) and (3,4);
• (2,2) and (3,4).

　　解析:

　　当时,题意明白的很快,但代码太暴力了,O(n^2)复杂度,而且n<=2e5,结果超时;

其实,这个不难想,每组寻找匹配的组时,只要分别去除了男女的重合序号,

但需注意的是:

1. 每组寻找时,会把所有的选一遍,所以最后合法的组数需要/2;
2. sum = sum + k - 男重 - 女重 + 1//最后+1是因为前面男重和女重都把自己减去了一遍,需要+1补齐;

上代码:

#include<iostream>

using namespace std;

typedef long long ll;

const int N = 2e5 + 10;

int main()
{
    int t;
    cin >> t;
    while(t --)
    {
        int a1[N], b1[N];
        int p, q, k, a[N], b[N];

        for(int i = 0; i < N; i ++)//初始化
            a1[i] = 0,b1[i] = 0;

        scanf("%d%d%d", &p , &q , &k);
        for(int i = 0; i < k; i ++)    scanf("%d", &a[i]),        a1[a[i]] ++;
        for(int i = 0; i < k; i ++)    scanf("%d", &b[i]),        b1[b[i]] ++;

        ll sum = 0;
        for(int i = 0; i < k; i ++)
            sum = sum + k - a1[a[i]] - b1[b[i]] + 1;
        printf("%lld\n", sum/2);
    }
    return 0;
}

//相比而言,开始这个太傻了 

//　　　　for(int i = 0; i < cou-1; i ++)
//　　　　　　for(int j = i+1; j < cou; j ++)
//　　　　　　　　if(a[i] != a[j] && b[i] != b[j])
//　　　　　　　　　　sum ++;