HZNUOJ 2508:双峰插云

 


我们注意到总的颜色种类只有200，故我们可以先处理出所有云朵中每一种颜色有多少朵云朵拥有，此处我们约定用$cnt[] 表示$

然后暴力枚举将哪朵云分给老薛，记录三个临时变量$Max_{kk}, Max_{xuexue}, pos$

$分别表示kk能够得到的最大颜色种类数，薛薛能够得到的最大颜色种类数，以及分给薛薛的云朵的下标$

枚举的时候注意要优先满足$kk, 如果当前得到的Max_{kk} 和之前的一样，再考虑一下薛薛$

$关于枚举的时候如果得到除了当前云朵外其他云朵的总的颜色总数$

首先记录一下预处理的$cnt[]中一共有多少个 >= 1 的元素，记为res, 此处用数组下标表示云朵标号$

$那么在枚举云朵的时候，我们用tmp表示除了这朵云外还剩下多少种不同颜色，显然tmp = res - x$

$x 表示有几种颜色是只有这朵云拥有的$

$再枚举这朵云拥有的所有颜色，判断在cnt[] 中这种颜色是不是 刚好为1即可$

$注意到一朵云的颜色种类数最多只有10.$

$总的时间复杂度是O(10n)$

#include <bits/stdc++.h>

using namespace std;

#define N 100010

int main() {
    int a[N][11];
    int b[N];
    int f[210];
    int n, m, i, j, t, p;
    scanf("%d%d",&n, &m);
    memset(f, 0, sizeof f);
    for (i = 1; i <= n; ++i) {
        scanf("%d",&a[i][0]);
        for (j = 1; j <= a[i][0]; ++j) {
            scanf("%d",&a[i][j]);
            f[a[i][j]]++;
        }
    }
    for (i = 1; i <= n; ++i) {
        b[i] = 0;
        for (j = 1; j <= a[i][0]; ++j)
            if (f[a[i][j]] == 1) 
                b[i]++;
    }
    t = 1;
    for (i = 2; i <= n; ++i)
        if (b[i] < b[t]) t = i;
        else if (b[i] == b[t] && a[i][0] > a[t][0]) t = i;
    puts("KK:");
    j = 1;
    while (j == t) ++j;
    printf("%d",j);
    for (i = j + 1; i <= n; ++i)
        if (i != t) 
            printf(" %d",i);
    printf("
");
    puts("Xuexue:");
    printf("%d
",t);
    return 0;
}