CF145C Lucky Subsequence 组合+dp

原题:http://codeforces.com/problemset/problem/145/C

题解:全部是4，7的数字被称为lucky数字，要求不出现重复lucky数字长度为k的集合个数，注：每个相同的lucky数字是不一样的。很容易发现若我们已经确定选i个lucky数字时剩下的数字可以随便选了.

设f(n)表示选n个不一样的lucky数字的方案数那么答案就是:。

(num为lucky数的种类，tot为非lucky数的个数)

问题就转化为怎么求f(i)，也可用dp的方法设:f[i][j]表示已经用了i个lucky，选j个方案数。

.cnt[i]为第i个lucky数出现的个数。

#include<bits/stdc++.h>
#define ll long long 
using namespace std;
const int M=1000000007;
const int N=1e5+10;
ll fac[N],inv[N];
ll f[2000][2000];
int n,k,num,tot;//lucky数的个数， 
int a[N],b[N],cnt[N];
ll qpow(ll a,ll b){
	ll ans=1;	
	while(b){
		if(b&1) ans=ans*a%M;
		b>>=1; a=a*a%M;
	}
	return ans;
}
inline void init(){
	fac[0]=1;int nn=1e5;
	for(int i=1;i<=nn;i++) fac[i]=fac[i-1]*i%M;
	inv[nn]=qpow(fac[nn],M-2);
	for(int i=nn-1;i>=0;i--) inv[i]=inv[i+1]*(i+1)%M;	
}

ll C(int n,int m){
	if(m>n) return 0;
	return fac[n]*inv[m]%M*inv[n-m]%M;
}
int main(){
	
	freopen("cf146.in","r",stdin);
	init();
	scanf("%d%d",&n,&k);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
	sort(a+1,a+n+1);
	num=0;b[num]=-1;
	for(int i=1;i<=n;i++){
		if(a[i]==b[num]){cnt[num]++;continue;} 
		int t=a[i];
		bool f=1;
		while(t){
			int temp=t%10;t/=10;
			if(temp!=4&&temp!=7){
				f=0;tot++;break;
			} 
		}
		if(f){b[++num]=a[i];cnt[num]=1;}
	}
	memset(f,0,sizeof f);
	for(int i=1;i<=num;i++) f[i][0]=1;
	for(int i=1;i<=num;i++) f[i][1]=cnt[i]+f[i-1][1];
	for(int j=2;j<=num;j++){
		for(int i=j;i<=num;i++){
			f[i][j]=(f[i-1][j-1]*cnt[i]+f[i-1][j])%M;
		}
	}
	ll ans=0;f[0][0]=1;
	for(int i=0;i<=num;i++){
		ans=(ans+f[num][i]*C(tot,k-i))%M;
	}
	printf("%lld\n",ans);  
	return 0;	
}