【分析】
将目之所及的题面抽象出一定的模型是每个优秀的OIer的基本能力。。。
先提供骗分50的优秀做法,显然的是要想使每个旅者可以回到自己的家乡,整一个交通网一定是由几个环组成的。题目要求在第k天回家,所以每个环的长度一定是k的约数。所以题目就改为,已知n和k求能否将n写成k的几个约数相加的形式(约数可以重复出现)。同时我们还可以发现所有的“约数”环可以拆成一定数量的“质数”环。这时问题就变得相当直白了。
50的时候我们可以把n作为背包容量,质数d作为物品体积,跑一遍完全背包。看体积n能否为1。再加上几种特判就能获得50分的好成绩了。
首先我们可以跑一遍预处理的欧拉筛,预处理出所有质数。接下来我们就要动用一些黑科技。因为k最多只有50种,我们可以对k进行记忆化,变相缩小T值。第一次运算时,先将k分解质因数。如果k只有2个质数,我们可以跑一遍扩展欧几里得,看是否有同时x和y为正的情况(存在,则是yes;不存在是no)。特殊情况,先判断n是否能为k的某个质数整除。如果只有唯一质因数且不可以整除,则一定是no。
最后考虑n>=3的情况。取k中最小的质因数(假设他叫p1),那由其他质因数相加结果mod p1的余数只会是0~p1-1。定义dist[i][j]为第i个k的其他质因数相加后mod p1为j的最小值。比如n%p1==dest,如果dist[i][dest]<=n,n一定可以写成a1*p1+(a2*p2+a3*p3+…+an*pn)的形式(任意a≥0)。
关于求余数的最小值我们可以转化为图论最短路,利用dijkstra求解。
具体看代码:
#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int maxq=55;//一组最多五十个k
const int maxlog=64;//每个k最多分解成log(k)个质数
const int maxn=100005;//最小质数的余数范围(最小质数)
const int maxp=2e6+5;//质数个数
const int maxv=3.2e7+5;//筛法范围
const LL INF=1ll<<60;
int q;LL memk[maxq];//第i号k的值
int tot,prime[maxp],fac[maxv];//筛法求质数,每个数的最大质因数
int cnt[maxq];LL p[maxq][maxlog];//第i号k分解的质因数个数,和每个质因数的值
LL dist[maxq][maxn];//第i号k的最小余数,用其他几个质数拼凑所产生的最小sum
struct info{LL dist; int home;};
priority_queue<info> heap;
bool vis[maxn];
bool operator <(info a,info b){
return a.dist>b.dist;
}
template<typename T> void read(T& x){
x=0;LL fl=1ll;char tmp=getchar();
while((tmp<'0'||tmp>'9')&&tmp!='-') tmp=getchar();
if(tmp=='-')fl=-fl,tmp=getchar();
while(tmp>='0'&&tmp<='9')x=(x<<1)+(x<<3)+tmp-'0',tmp=getchar();
x*=fl;
return ;
}
void exgcd(LL a,LL b,LL &x,LL &y){
if(b==0){
x=1,y=0;
return ;
}
exgcd(b,a%b,y,x);
y-=a/b*x;
return ;
}
int Num;
int main(){
for(int i=2;i<maxv;i++){//这里有一个跑得飞快的欧拉筛
if(!fac[i]) prime[++tot]=fac[i]=i;
for(int j=1;j<=tot&&prime[j]<=fac[i];j++){
int tmp=prime[j]*i;
if(tmp>=maxv) break;
fac[tmp]=prime[j];
}
}
// read(Num);
scanf("%d",&Num);
int T;read(T);
while(T--){
LL n,k;
read(n),read(k);
int pos=0;
for(int i=1;i<=q;i++)//记忆化
if(memk[i]==k)pos=i;
if(pos==0){
pos=++q;
memk[q]=k;
long long tmp=k;
for(int i=1;1ll*prime[i]*prime[i]<=tmp;i++){
if(tmp%prime[i]==0){
p[pos][++cnt[pos]]=prime[i];
while(tmp%prime[i]==0)tmp/=prime[i];
}
}//将k分解质因数
if(tmp!=1) p[pos][++cnt[pos]]=tmp;
if(cnt[pos]>=3){
for(int i=0;i<p[pos][1];i++)
dist[pos][i]=INF;
dist[pos][0]=0;heap.push((info){0,0});
memset(vis,false,sizeof(vis));
while(!heap.empty()){//看我用dijkstra跑sum值
while(!heap.empty()&&vis[heap.top().home])heap.pop();
if(heap.empty()) break;
info tmp=heap.top();heap.pop();
for(int i=2;i<=cnt[pos];i++){
int dest=(tmp.home+p[pos][i])%p[pos][1];
if(dist[pos][dest]>tmp.dist+p[pos][i]){
dist[pos][dest]=tmp.dist+p[pos][i];
heap.push((info){dist[pos][dest],dest});
}
}
}
}
}
bool flag=false;
for(int i=1;i<=cnt[pos];i++)
if(n%p[pos][i]==0){
flag=true;
printf("YES\n");
break;
}
if(flag) continue;
if(cnt[pos]<=1){
printf("NO\n");
continue;
}
if (cnt[pos] == 2) {
LL x = 0, y = 0;
exgcd(p[pos][1], p[pos][2], x, y);
y = (y % p[pos][1] + p[pos][1]) % p[pos][1];
LL tmp = y * (n % p[pos][1]) % p[pos][1] * p[pos][2];
if (tmp <= n) printf("YES\n");
else printf("NO\n");
continue;
}
int tmp=n%p[pos][1];
if(dist[pos][tmp]<=n)printf("YES\n");
else printf("NO\n");
}
return 0;
}