bzoj 4524 [Cqoi2016]伪光滑数

搜索。
首先将所有小于128的质数预处理出来,总共31个。设第i个质数为 p[i]
用一个结构体 {V,a[31]} 来表示一个数字 V ,将它分解后有 a[i] p[i]
然后求出所有 p[i]j(i[1,31],j1&&p[i]jN) ,按照上面结构体的方式构造后加入一个以数字大小为关键字的大根堆中。
之后重复 K1 次操作,每次操作都是取出堆顶,将堆顶元素的某个质因子 p[i] 变为 p[i1] ,将 p[i]/p[i1] 后再加入堆中。用哈希表判重,保证加入堆中的数字不重复。
第K次操作时的堆顶就是答案。
由于每次操作最多使堆中多加入30个元素,所以时间复杂度为 O(30klog(30k))
代码:

#include
#include
#include
#include
using namespace std;
typedef long long LL;
const LL Mod=9999991;
LL p[100005],sz,h[10000005],cnt=0;
struct front_star{LL ne,to,a[32];}a[1000005];
priority_queueq;
void Addedge(LL x,LL y)
{   a[++cnt]=a[0];a[cnt].to=y;a[cnt].ne=h[x];h[x]=cnt;}
LL find(LL V)
{   LL x=V%Mod,k;
    for(k=h[x];k;k=a[k].ne)if(V==a[k].to)return k;
    return 0;
}
bool ck(LL V)
{   if(find(V))return true;
    else return false;
}
void mk(LL V)
{   LL x=V%Mod;
    Addedge(x,V);
}
int main()
{   LL N,K,i,j,k,bj,ans,ans2;
    a[0].ne=a[0].to=0;
    for(i=0;i<32;i++)a[0].a[i]=0;
    for(i=2;i<=128;i++)
      {bj=0;
       for(j=2;jif(i%j==0){bj=1;break;}}
       if(!bj)p[++p[0]]=i;
      }
    cin>>N>>K;
    for(i=1;p[i]<=N&&i<=p[0];i++)
      {for(j=1,ans=p[i];ans<=N&&ans>0;j++,ans*=p[i])
         {mk(ans);
          a[cnt].a[i]=j;
          q.push(ans);sz++;
         }
      }
    while(K--)
      {ans=q.top();q.pop();
       if(!K){cout<return 0;}
       k=find(ans);
       for(i=2;i<=31;i++)
         {if(a[k].a[i])
            {ans2=ans/p[i]*p[i-1];
             if(ck(ans2)==false)
               {q.push(ans2);sz++;mk(ans2);
                for(j=1;j<32;j++)a[cnt].a[j]=a[k].a[j];
                a[cnt].a[i]--;a[cnt].a[i-1]++;
               }
            }
         }
      }
    return 0;
}

你可能感兴趣的:(各省省选)