[Luogu P3298] [BZOJ 3198] [SDOI2013]泉

洛谷传送门

BZOJ传送门

题目描述

作为光荣的济南泉历史研究小组中的一员，铭铭收集了历史上$x$个不同年份时不同泉区的水流指数，这个指数是一个小于$2^{30}$的非负整数。第$i$个年份时六个泉区的泉水流量指数分别为 $A(i,1)，A(i，2)，A(i，3)，A(i，4)，A(i，5)$与 $A(i，6)$。

现在铭铭希望知道有多少对不同的年份：$i$和$j$,满足这两年恰好有$K$个泉区的泉水流$S$指数对应相同。

输入输出格式

输入格式：

第一行有$2$个整数，分别是$N$和$K$

之后$N$行，每行有$6$个整数。第$i$行的第$j$个数字$A(i，j)$表示第$i$个年份中第$j$个泉区的泉水流量指数。

输出格式：

一个整数表示有多少对不同的年份满足恰有$K$个区的泉水流量指数对应相同。

输入输出样例

输入样例#1：

3 3
1 2 3 4 5 6
1 2 3 0 0 0
0 0 0 4 5 6

输出样例#1：

2

说明

对于 100%的数据, $0\le K\le 6$, 且所有数据中$K$是等概率出现的, 即对于任意的 $0\le x$都有大约 $1/7$ 的数据中 $K=x$. $N\le 100000$

解题分析

很显然的容斥， 我们先可以用哈希算出来至少有$K$个，$K+1$…$6$个相同的对数， 设这些值为$f(K),f(K+1)...f(6)$。再设$g(K),g(K+1),...,g(6)$为恰好这么多相同的对数

显然， $g(K+1)$对$f(K)$的贡献是$\left(\genfrac{}{}{0ex}{}{K+1}{K}\right)$次， 对$g(K+2)$的贡献是$\left(\genfrac{}{}{0ex}{}{K+2}{K}\right)$次… ，大概是这样的：

	$f(K)=$	$f(K+1)=$	$f(K+2)=$
$g(K)$	$\left(\genfrac{}{}{0ex}{}{K}{K}\right)$
$g(K+1)$	$\left(\genfrac{}{}{0ex}{}{K+1}{K}\right)$	$\left(\genfrac{}{}{0ex}{}{K+1}{K+1}\right)$
$g(K+2)$	$\left(\genfrac{}{}{0ex}{}{K+2}{K}\right)$	$\left(\genfrac{}{}{0ex}{}{K+2}{K+1}\right)$	$\left(\genfrac{}{}{0ex}{}{K+2}{K+2}\right)$

这是不能直接容斥的， 但观察可得每一列的下指标一样， 那么其中每一项乘上$\left(\genfrac{}{}{0ex}{}{K+m}{K}\right)$后， 就变成了：

	$K$	$K+1$	$K+2$
$K$	$\left(\genfrac{}{}{0ex}{}{K}{K}\right)\left(\genfrac{}{}{0ex}{}{0}{0}\right)$
$K+1$	$\left(\genfrac{}{}{0ex}{}{K+1}{K}\right)\left(\genfrac{}{}{0ex}{}{1}{0}\right)$	$\left(\genfrac{}{}{0ex}{}{K+1}{K}\right)\left(\genfrac{}{}{0ex}{}{1}{1}\right)$
$K+2$	$\left(\genfrac{}{}{0ex}{}{K+2}{K}\right)\left(\genfrac{}{}{0ex}{}{2}{0}\right)$	$\left(\genfrac{}{}{0ex}{}{K+2}{K}\right)\left(\genfrac{}{}{0ex}{}{2}{1}\right)$	$\left(\genfrac{}{}{0ex}{}{K+2}{K}\right)\left(\genfrac{}{}{0ex}{}{2}{2}\right)$

就可以容斥了…

auto真好用

代码如下：

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cctype>
#include <algorithm>
#include <cmath>
#include <vector>
#define R register
#define IN inline
#define W while
#define gc getchar()
#define BASE 666233ll
#define MOD1 100003ll
#define ll long long
#define MOD2 19260817ll
#define MX 100500
template <class T>
IN void in(T &x)
{
    x = 0; R char c = gc;
    for (; !isdigit(c); c = gc);
    for (;  isdigit(c); c = gc)
    x = (x << 1) + (x << 3) + c - 48;
}
struct INFO {int key, cnt;};
std::vector <INFO> buc[MOD1];
int n, k;
long long ans, res;
int dat[MX][6], C[7][7], tar[7];
IN void insert(ll key1, ll key2)
{
    for (auto &i : buc[key1]) if (i.key == key2) return i.cnt++, void();
    buc[key1].push_back({key2, 1});
}
IN void count()
{
    for (R int i = 0; i < MOD1; ++i)
    {
        for (auto j : buc[i]) res += 1ll * j.cnt * (j.cnt - 1) / 2;
        std::vector <INFO> ().swap(buc[i]);
    }
}
IN void work(R int typ)
{
    int tot = 0; ll hs1, hs2;
    for (R int i = 0; i < 6; ++i) if (typ & (1 << i)) tar[++tot] = i;
    for (R int i = 1; i <= n; ++i)
    {
        hs1 = hs2 = 0;
        for (R int j = 1; j <= tot; ++j)
        {
            hs1 = (hs1 * BASE % MOD1 + dat[i][tar[j]]) % MOD1;
            hs2 = (hs2 * BASE % MOD2 + dat[i][tar[j]]) % MOD2;
        }
        insert(hs1, hs2);
    }
    count();
}
int main(void)
{
    in(n), in(k);
    for (R int i = 1; i <= n; ++i)
    for (R int j = 0; j <  6; ++j) in(dat[i][j]);
    for (R int i = 0; i <= 6; ++i) C[i][0] = 1;
    for (R int i = 1; i <= 6; ++i)
    for (R int j = 1; j <= i; ++j)
    C[i][j] = C[i - 1][j - 1] + C[i - 1][j];
    for (int i = 0; i <= 63; ++i)
    {
        int cnt = __builtin_popcount(i);
        if (cnt < k) continue;
        res = 0; work(i);
        if ((k - cnt) & 1) ans -= res * C[cnt][k];
        else ans += res * C[cnt][k];
    }
    printf("%lld", ans);
}