====== 同余最短路 ======
===== 算法简介 =====
用于计算 $k_1a_1+k_2a_2+\cdots +k_na_n$ 在 $[0,m]$ 范围内能表示的数的算法。
===== 算法实现 =====
考虑建点 $0,1\cdots a_1-1$。然后对每个点 $i$,连边 $i\to (i+a_j)\bmod a_1(w=a_j)$,再跑最短路。
于是可以 $O(n\times a\log a)$ 计算出最小的可以表示成 $ka_1+r(0\le r\lt a_1)$ 的数,于是每个 $r$ 对答案的贡献为 $\lfloor \frac {m-\text{dis}(r)}{a_1}\rfloor$。
不难发现可以重新排序选最小的 $a$ 作为 $a_1$,另外每个点的相邻点可以在跑最短路算法时动态计算,这些都有利于卡常和节省空间。
===== 算法例题 =====
==== 例题一 ====
[[https://www.luogu.com.cn/problem/P2371|洛谷p2371]]
板子题。
const int MAXN=15,MAXV=5e5+5;
const LL inf=1e18;
int a[MAXN];
LL dis[MAXV];
bool vis[MAXV];
void dj(int n){
priority_queue >q;
q.push(make_pair(0,0));
_for(i,1,a[0])
dis[i]=inf;
while(!q.empty()){
int u=q.top().second;q.pop();
if(vis[u])
continue;
vis[u]=true;
_for(i,1,n){
int v=(u+a[i])%a[0],w=a[i];
if(dis[v]>dis[u]+w){
dis[v]=dis[u]+w;
q.push(make_pair(-dis[v],v));
}
}
}
}
LL calc(LL val){
LL ans=0;
_for(i,0,a[0]){
if(dis[i]<=val)
ans+=(val-dis[i])/a[0]+1;
}
return ans;
}
int main(){
int n=read_int();
LL ql=read_LL(),qr=read_LL();
_for(i,0,n)
a[i]=read_int();
sort(a,a+n);
dj(n);
enter(calc(qr)-calc(ql-1));
return 0;
}
==== 例题二 ====
[[http://acm.hdu.edu.cn/showproblem.php?pid=6071|HDU6071]]
=== 题意 ===
给定一个四元环,环上每条边有一个边权,且重复经过则计算多次贡献。
人物从二号点出发,最终回到二号点,问权值不小于 $k$ 的路径的最小权值。
=== 题解 ===
任取一条与二号点相邻的边,设权值为 $w$。设 $\text{dis}(i,j)$ 表示从二号点出发到达 $i$ 号点且距离模 $2w$ 等于 $j$ 的最小距离。
那么从二号点出发能得到的路径的权值集合为 $\{\text{dis}(2,r)+2kw|0\le r\lt 2w,k\ge 0\}$,枚举 $r$ 即可计算答案。
关于为什么选取 $2w$,可以解释为回到 $2$ 号点后可以在任意一条边上来回移动,正确性不难证明。
const int MAXN=5,MAXV=6e4+5;
const LL inf=2e18;
int d[MAXN];
vector > g[MAXN];
LL dis[MAXN][MAXV];
bool vis[MAXN][MAXV];
void dj(int n){
priority_queue > q;
q.push(make_pair(0,1));
_for(i,0,4)_for(j,0,n)
dis[i][j]=inf,vis[i][j]=false;
dis[1][0]=0;
while(!q.empty()){
int u1=q.top().second;
int u2=(-q.top().first)%n;
q.pop();
if(vis[u1][u2])
continue;
vis[u1][u2]=true;
for(pair p:g[u1]){
int v1=p.first,w=p.second,v2=(u2+w)%n;
if(dis[v1][v2]>dis[u1][u2]+w){
dis[v1][v2]=dis[u1][u2]+w;
q.push(make_pair(-dis[v1][v2],v1));
}
}
}
}
LL calc(int val,LL k){
LL ans=inf;
_for(i,0,val){
if(dis[1][i]>k)
ans=min(ans,dis[1][i]);
else
ans=min(ans,dis[1][i]+(k-dis[1][i]+val-1)/val*val);
}
return ans;
}
void solve(){
LL k=read_LL();
_for(i,0,4)
g[i].clear();
_for(i,0,4){
d[i]=read_int();
g[i].push_back(make_pair((i+1)%4,d[i]));
g[(i+1)%4].push_back(make_pair(i,d[i]));
}
dj(2*d[0]);
enter(calc(2*d[0],k));
}
int main(){
int T=read_int();
while(T--)
solve();
return 0;
}
==== 例题三 ====
[[https://www.luogu.com.cn/problem/CF986F|CF986F]]
=== 题意 ===
$T$ 组询问,每组询问给定 $n,k$,求 $n$ 是否能用若干个 $k$ 的不为 $1$ 的因子表示。
数据保证 $k$ 的取值不超过 $50$ 个。
=== 题解 ===
如果 $n$ 能用若干个 $k$ 的不为 $1$ 的因子表示等价于 $n$ 能用 $k$ 的素因子表示。
预处理 $O(\sqrt k)$ 范围的数后对每个 $k$ 进行因式分解,时间复杂度 $O\left(\frac {\sqrt k}{\ln k}\right)$。
如果 $k=1$,显然无解。如果 $k$ 是质数,只需要判定 $k\mid n$ 是否成立。
如果 $k$ 有两种素因子,记为 $a,b$,于是等价于求解 $ax+by=n$ 的非负整数解。
找到最小的 $y$ 满足 $y\equiv nb^{-1}\bmod x$,然后验证此时 $by$ 是否不超过 $n$ 即可。
如果 $k$ 有至少三种素因子,显然 $k$ 的最小素因子不超过 $O\left(k^{\frac 13}\right)$,跑同余最短路算法即可,时间复杂度 $O\left(k^{\frac 13}\log k\right)$。
总时间复杂度 $\left(50\left(\frac{\sqrt k}{\ln k}+k^{\frac 13}\log k\right)\right)$。
const int MAXV=4e7,MAXN=1e5+5,MAXQ=1e4+5;
const LL inf=1e18+5;
int prime[MAXV],pcnt;
bool vis[MAXV];
void Init(){
_for(i,2,MAXV){
if(!vis[i])prime[pcnt++]=i;
for(int j=0;j g;
void get_frac(LL n){
g.clear();
_for(i,0,pcnt){
if(n%prime[i]==0){
g.push_back(prime[i]);
while(n%prime[i]==0)n/=prime[i];
}
}
if(n!=1)
g.push_back(n);
}
int quick_pow(int n,int k,int mod){
int ans=1;
while(k){
if(k&1)ans=1LL*ans*n%mod;
n=1LL*n*n%mod;
k>>=1;
}
return ans;
}
LL dis[MAXN];
void dj(int n){
_for(i,0,n){
dis[i]=inf;
vis[i]=false;
}
dis[0]=0;
priority_queue q;
q.push(0);
while(!q.empty()){
int u=(-q.top())%n;q.pop();
if(vis[u])
continue;
vis[u]=true;
_for(i,1,g.size()){
int v=(u+g[i])%n;
if(dis[v]>dis[u]+g[i]){
dis[v]=dis[u]+g[i];
q.push(-dis[v]);
}
}
}
}
bool query(LL n){
if(g.size()==0)
return false;
if(g.size()==1)
return n%g[0]==0;
else if(g.size()==2){
LL x=g[0],y=g[1];
int b=(n%x)*quick_pow(y%x,x-2,x)%x;
return y*b<=n;
}
else{
int m=g[0];
return dis[n%m]<=n;
}
}
map > >mp;
bool ans[MAXQ];
void solve(LL k,vector > a){
get_frac(k);
if(g.size()>2)
dj(g[0]);
for(pair p:a)
ans[p.first]=query(p.second);
}
int main(){
Init();
int q=read_int();
_for(i,0,q){
LL n=read_LL(),k=read_LL();
mp[k].push_back(make_pair(i,n));
}
for(map > >::iterator iter=mp.begin();iter!=mp.end();iter++)
solve(iter->first,iter->second);
_for(i,0,q){
if(ans[i])
puts("YES");
else
puts("NO");
}
return 0;
}