BZOJ3131: [Sdoi2013]淘金（洛谷P3303）

数位DP 贪心

BZOJ题目传送门
洛谷题目传送门

很显然最后有金子的格子坐标一定可以分解成$2,3,5,7$的幂次。暴枚后发现只有$14672$个。那么离散化之后就可以DP求出每个数字一共出现多少次了。

因为这是二维坐标，那么就相当于一个序列中取前$k$大的数对，每个数对的贡献为$w[i]\ast w[j]$。可以排序后对每一个位置维护一个指针指向当前选择的数对，每次选择当前数对后把指针往后移一位。用堆维护即可。

代码：

#include<queue>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 15000
#define F inline
using namespace std;
typedef long long LL;
const LL MOD=1e9+7;
struct node{ int x; LL w; };
LL n,m,k,ans,a[N],b[20],c[N],id[N],f[20][N][2];
priority_queue <node> q;
F bool operator < (node a,node b){ return a.w<b.w; }
F bool cmp(LL x,LL y){ return x>y; }
F void Make(){
	LL p2[50],p3[50],p5[50],p7[50],mx=n;
	p2[0]=p3[0]=p5[0]=p7[0]=1;
	while (mx) b[++b[0]]=mx%10,mx/=10;
	for (int i=1,f2=1,f3=1,f5=1,f7=1;i<=50;i++){
		if (f2) p2[i]=p2[i-1]*2; if (f3) p3[i]=p3[i-1]*3;
		if (f5) p5[i]=p5[i-1]*5; if (f7) p7[i]=p7[i-1]*7;
		if (f2&&p2[i]>n) f2=0; if (f3&&p3[i]>n) f3=0;
		if (f5&&p5[i]>n) f5=0; if (f7&&p7[i]>n) f7=0;
	}
	for (int i=0;p2[i]<=n;i++)
	for (int j=0;p2[i]*p3[j]<=n;j++)
	for (int k=0;p2[i]*p3[j]*p5[k]<=n;k++)
	for (int p=0;p2[i]*p3[j]*p5[k]*p7[p]<=n;p++)
		a[++m]=p2[i]*p3[j]*p5[k]*p7[p];
	a[++m]=0,sort(a+1,a+m+1),f[0][2][0]=1;
	m=unique(a+1,a+m+1)-a-1;
}
int main(){
	scanf("%lld%lld",&n,&k),Make();
	for (int i=0;i<=b[0];i++)
	for (int j=1;j<=m;j++)
	for (int k=0;k<2;k++)
	if (f[i][j][k])
		for (int x=(i!=0),y;x<=9;x++){
			y=lower_bound(a+1,a+m+1,a[j]*x)-a;
			f[i+1][y][(bool)(k+x>b[i+1])]+=f[i][j][k];
		}
	for (int j=1;j<=m;j++)
		for (int i=1;i<=b[0];i++)
			c[j]+=f[i][j][0]+(i!=b[0])*f[i][j][1];
	sort(c+1,c+m+1,cmp);
	for (int i=2;i<=m;i++) id[i]=2;
	for (int i=2;i<=m;i++) q.push((node){i,c[i]*c[2]});
	while (!q.empty()&&k){
		LL w=q.top().w; int x=q.top().x;
		q.pop(),(ans+=w)%=MOD,k--;
		if (id[x]<m) q.push((node){x,c[x]*c[++id[x]]});
	}
	return printf("%lld\n",ans),0;
}