首页 > 代码库 > BZOJ 3131 淘金
BZOJ 3131 淘金
考虑到乘出来的东西实际上不多。。。直接map记下。
然后比如说x位的计数,就把ceil(x/2)和trunc(x/2)的情况乘起来。
然后就是一个ai,j=tab[i]*tab[j],求这个数表前k大的问题。
这个可以排序,然后单调队列,把状态慢慢往后推就行了。(注意到ai,j都不用取模才可以这么做,否则。。。我也不知道怎么做)
#include<iostream> #include<cstdio> #include<cstring> #include<algorithm> #include<map> #include<queue> #include<cmath> #define maxn 1000050 #define mod 1000000007 using namespace std; long long n,k,bit[20],ret=0,a[maxn],tot=0; long long ans=0; map <long long,long long> tab[7],mp; map <pair<long long,long long>,long long> vis; map <long long,long long>::iterator it1,it2; struct status { long long p1,p2; long long val; status (long long p1,long long p2,long long val):p1(p1),p2(p2),val(val) {} status () {} friend bool operator < (const status &x,const status &y) { return x.val<y.val; } }; priority_queue <status> q; bool cmp(long long x,long long y) {return x>y;} void pre_dfs(long long now,long long ret) { if (now-1) tab[now-1][ret]++; if (now==7) return; for (long long i=1;i<=9;i++) pre_dfs(now+1,ret*i); } void get_bit(long long x) { ret=0; while (x) {bit[++ret]=x%10;x/=10;} } void update(long long ret,long long x) { if (!x) {mp[ret]++;return;} if (!(x-1)) {for (long long i=1;i<=9;i++) mp[ret*i]++;return;} long long l=ceil((double)x/2),r=trunc((double)x/2); for (it1=tab[l].begin();it1!=tab[l].end();it1++) for (it2=tab[r].begin();it2!=tab[r].end();it2++) mp[ret*(it1->first)*(it2->first)]+=it1->second*it2->second; } void digit_dp() { for (long long i=1;i<=ret-1;i++) update(1,i); long long rets=1; for (long long i=ret;i>=1;i--) { for (long long j=1;j<bit[i];j++) update(rets*j,i-1); rets*=bit[i]; } mp[rets]++; } int main() { scanf("%lld%lld",&n,&k); pre_dfs(1,1); get_bit(n); digit_dp(); for (it1=mp.begin();it1!=mp.end();it1++) { if (!it1->first) continue; a[++tot]=it1->second; } sort(a+1,a+tot+1,cmp); q.push(status(1,1,(long long)a[1]*a[1])); long long tr=0; for (long long i=1;i<=k;i++) { tr++; status now=q.top();q.pop(); if (vis[make_pair(now.p1,now.p2)]) {i--;continue;} ans=(ans+now.val%mod)%mod;vis[make_pair(now.p1,now.p2)]=1; q.push(status(now.p1,now.p2+1,(long long)a[now.p1]*a[now.p2+1])); q.push(status(now.p1+1,now.p2,(long long)a[now.p1+1]*a[now.p2])); } printf("%lld\n",ans%mod); return 0; }
BZOJ 3131 淘金
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。