bzoj 4524: [Cqoi2016]伪光滑数

题意：

若一个大于1的整数M的质因数分解有k项，其最大的质因子为Ak，并且满足Ak^K<=N,Ak<128，我们就称整数M为N-伪光滑数。现在给出N，求所有整数中，第K大的N-伪光滑数。

题解：

好题，只能%：beginend
话说下次遇到这种第k大是不是也可以这么搞啊
code：

#include
#include
#include
#include
#include
#include
#define LL long long
using namespace std;
int prime[50],pr=0;
bool v[130];
struct trnode{
    int lc,rc,d;
    LL c,t;
}tr[17000005];int tot=0;
int g[50][110],f[50][110];
LL n;int K;
void pre()
{
    memset(v,true,sizeof(v));
    for(int i=2;i<128;i++)
        if(v[i])
        {
            prime[++pr]=i;
            for (int j=i*2;j<128;j+=i) v[j]=false;
        }
}
int new_node(int x,LL t)
{
    tot++;tr[tot]=tr[x];
    tr[tot].c*=t;tr[tot].t*=t;
    return tot;
}
void pushdown(int x)
{
    if(tr[x].t==1) return;
    LL t=tr[x].t;tr[x].t=1;
    if(tr[x].lc) tr[x].lc=new_node(tr[x].lc,t);
    if(tr[x].rc) tr[x].rc=new_node(tr[x].rc,t);
}
int merge(int x,int y)
{
    if(!x||!y) return x+y;
    pushdown(x);pushdown(y);
    if(tr[x].c<tr[y].c) swap(x,y);
    int v=new_node(x,1);tr[v].rc=merge(tr[v].rc,y);
    if(tr[tr[v].lc].d<tr[tr[v].rc].d) swap(tr[v].lc,tr[v].rc);
    tr[v].d=tr[tr[v].rc].d+1;
    return v;
}
priority_queueint,int> > > q;
void make_heap()
{
    pre();
    g[0][0]=++tot;tr[tot].c=tr[tot].t=1;
    for(int i=1;i<=pr;i++)
    {
        g[i][0]=1;
        for(LL j=1,p=prime[i];p>0&&p<=n;j++,p*=prime[i])
        {
            for(LL k=1,pri=prime[i];k<=j;k++,pri*=prime[i]) f[i][j]=merge(f[i][j],new_node(g[i-1][j-k],pri));
            g[i][j]=merge(g[i-1][j],f[i][j]);
            q.push(make_pair(tr[f[i][j]].c,make_pair(i,j)));
        }
    }
}
void solve()
{
    LL ans;
    while(K--)
    {
        ans=q.top().first;int i=q.top().second.first,j=q.top().second.second;q.pop();
        pushdown(f[i][j]);f[i][j]=merge(tr[f[i][j]].lc,tr[f[i][j]].rc);
        q.push(make_pair(tr[f[i][j]].c,make_pair(i,j)));
    }
    printf("%lld",ans);
}
int main()
{
    scanf("%lld %d",&n,&K);
    make_heap();solve();
}