【Atcoder abc246_f】Typewriter【状态压缩】【组合数学，容斥】

分析

计数题，DP状态都存不下，考虑组合数学。

先用单个字符串手摸一下，发现一个长度为k的字符串打出L个字符最多有 $k^L$ 种。答案显然不可能是简单相加。于是我又想到了去重，但是不是一个字符串直接去重肯定是不行。那么数学方法又有什么能达到“求并集，去重”的效果呢？容斥原理！

n个集合的容斥原理公式参考百度就非常清楚了，发现是偶数个集合的交集前面就是减号，反之则为加号。

注意到n的值特别小，仿佛跟状压DP的一样，于是考虑存储所有字母存在性的状态，一共 $2^{26}$ 完全可以枚举和存储。

所以最后的做法就是输入时统计每个字符串有的字母（那一位设为1），然后枚举所有的状态（也就是涵盖了需要容斥的所有集合状态），对于每一种状态，都枚举所有字符串求交集，看看有多少个字母（多少个1），然后答案就根据集合个数的奇偶性判断增减。

上代码

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
typedef long long ll;

const int mod=998244353;
ll n,k,v[20],ans;

ll ksm(ll x,ll y)
{
	ll sum=1;
	while(y)
	{
		if(y&1) sum=sum*x%mod;
		x=x*x%mod;
		y>>=1;
	}
	return sum%mod;
}

int main()
{
	cin>>n>>k;
	for(int i=1;i<=n;i++)
	{
		string s;
		cin>>s;
		int len=s.length();
		for(int j=0;j<len;j++)
		{
			v[i]=v[i]|(1<<(s[j]-'a'));
		}
	}
	for(int i=1;i<=(1<<n)-1;i++)
	{
		ll tot=(1<<26)-1;
		for(int j=1;j<=n;j++)
		{
			if(i&(1<<(j-1))) tot&=v[j];
		}
		//这个函数可以直接求二进制中1的个数！
		if(__builtin_popcount(i)&1) ans=(ans+ksm(__builtin_popcount(tot),k))%mod;
		else ans=(ans-ksm(__builtin_popcount(tot),k)+mod)%mod;
	}
	cout<<ans%mod;
	return 0;
}