bzoj3198

容斥原理+哈希表

恰好k个，那么上容斥原理，我们先2^6枚举相同的位置，用哈希表判断有多少个对应位置相同的元素，然后用容斥原理计算，似乎这里的容斥没有那么简单，详见这里 http://www.cnblogs.com/candy99/p/mobius.html， 要乘上组合数计算

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
typedef long long ll;
const int N = , mod = ;
int n, k;
ll ans;
int c[][], a[N][];
namespace Hash_List {
    int cnt, tot;
    int h[N], bel[N];
    struct List {
        int nxt, id, cnt;
    } e[N << ];
    void ini() { ++cnt; }
    bool same(int x, int y, int s) {
        for(int i = ; i < ; ++i) if(s & ( << i) && a[x][i] != a[y][i]) return false;
        return true;
    }
    int hash(int *a, int s) {
        int h = ;
        for(int i = ; i < ; ++i) if(s & ( << i)) h = ((h *  % mod + a[i] % mod) % mod + mod) % mod;
        return h;
    }
    int insert(int id, int s) {
        int val = hash(a[id], s);
        if(bel[val] != cnt) h[val] = , bel[val] = cnt; // 如果这个没有被标记过，那么就暴力查找，否则直接插入
        for(int i = h[val]; i; i = e[i].nxt) if(same(e[i].id, id, s)) { ++e[i].cnt; return e[i].cnt - ; }
        e[++tot] = (List) {h[val], id, }, h[val] = tot;
        return ;
    }
} using Hash_List :: insert; using Hash_List :: ini;
int one(int x) {
    int ret = ;
    while(x) ++ret, x &= (x - );
    return ret;
}
int main()
{
    scanf("%d%d", &n, &k);
    c[][] = ;
    for(int i = ; i <= ; ++i)
    {
        c[i][] = ;
        for(int j = ; j <= i; ++j) c[i][j] = c[i - ][j] + c[i - ][j - ];
    }
    for(int i = ; i <= n; ++i)
        for(int j = ; j < ; ++j) scanf("%d", &a[i][j]);
    for(int i = ; i < ; ++i)
    {
        int t = one(i);
        if(t < k) continue;
        Hash_List :: ini();
        for(int j = ; j <= n; ++j) ans += (((t - k) & ) ? - : ) * (ll)Hash_List :: insert(j, i) * (ll)c[t][k];
    }
    printf("%lld\n", ans);
    return ;
}