题目链接:https://atcoder.jp/contests/abc077/tasks/arc084_b?lang=en
题意:给一个
k
k
k,对于所有
k
k
k的倍数,求这些数中十进制下各位和的最小值。
思路:首先考虑暴力枚举
k
k
k的倍数,由于上界不好确定,而且答案对应的数字可能很大,不太可行。
k的倍数可以看作模k余0的同余类,对于任意一个数,都可以由1,不断进行+1,*10操作构造出来的,在这个过程中,+1会给答案增加1的权值,*10不会给答案带来权值;所以在模k的完全剩余系中对应元素连边(对于一个
x
x
x,我们给
x
+
1
%
k
x+1 \% k
x+1%k和
10
∗
x
%
k
10*x \% k
10∗x%k),求最短路,答案就是1到0的最短路权值+1(出发点选1可以保证对于任意k 它模k的值都相同 )。
代码:
#include<bits/stdc++.h>
using namespace std;
typedef pair<int,int> pii;
const int N = 1e6 + 10;
int n, m;
int h[N], w[N], e[N], ne[N], idx;
int dist[N];
bool st[N];
void add(int a, int b, int c)
{
e[idx] = b, w[idx] = c, ne[idx] = h[a], h[a] = idx ++ ;
}
//最短路这里好像可以用双端队列优化到O(n)
void dijkstra()
{
memset(dist,0x3f,sizeof dist);
dist[1]=0;
priority_queue<pii,vector<pii>,greater<pii>> heap;
heap.push({0,1});
int x=0;
while(heap.size())
{
auto t=heap.top();
heap.pop();
if(st[t.second]) continue;
st[t.second]=true;
for(int i=h[t.second];i!=-1;i=ne[i])
{
int j=e[i];
if(dist[j]>t.first+w[i])
{
dist[j]=t.first+w[i];
heap.push({dist[j],j});
}
}
}
}
int main()
{
int k;
scanf("%d",&k);
memset(h,-1,sizeof(h));
//0->1的边就不加了,不会走到的,加了还多一个环
for(int i=1;i<k;i++)
{
if(i%k!=(i+1)%k) add(i%k,(i+1)%k,1);
if(i%k!=(10*i)%k) add(i%k,(10*i)%k,0);
}
dijkstra();
printf("%d\n",1+dist[0]);
}
双端队列O(N)最短路
#include<bits/stdc++.h>
using namespace std;
typedef pair<int,int> pii;
const int N=2e5+7,inf=0x3f3f3f3f;
int n, m;
int h[N],w[N],e[N],ne[N],idx;
int dist[N];
bool st[N];
void add(int a, int b, int c)
{
e[idx] = b, w[idx] = c, ne[idx] = h[a], h[a] = idx ++ ;
}
//1到0的最短路
void bfs()
{
dist[1]=0;
deque<int> q;
q.push_back(1);
while(q.size())
{
int u=q.front();q.pop_front();
if(st[u]) continue;
st[u]=true;
for(int i=h[u];i!=-1;i=ne[i])
{
int v=e[i];
if(w[i]==0) q.push_front(v);
else q.push_back(v);
if(dist[v]>dist[u]+w[i]) dist[v]=dist[u]+w[i];
}
}
}
int main()
{
int t;
scanf("%d",&t);
// t=1;
while(t--)
{
int k;
scanf("%d",&k);
memset(h,-1,sizeof h);
memset(st,0,sizeof st);
memset(dist,0x3f,sizeof dist);
idx=0;
for(int i=0;i<k;i++)
{
if(i%k!=(i+1)%k) add(i%k,(i+1)%k,1);
if(i%k!=(10*i)%k) add(i%k,(10*i)%k,0);
}
bfs();
printf("%d\n",1+dist[0]);
}
}