[AtCoder Beginner Contest 077]D-Small Multiple (思维、最短路)

题目链接：https://atcoder.jp/contests/abc077/tasks/arc084_b?lang=en
题意：给一个$k$，对于所有$k$的倍数，求这些数中十进制下各位和的最小值。
思路：首先考虑暴力枚举$k$的倍数，由于上界不好确定，而且答案对应的数字可能很大，不太可行。
k的倍数可以看作模k余0的同余类，对于任意一个数，都可以由1，不断进行+1，*10操作构造出来的，在这个过程中，+1会给答案增加1的权值，*10不会给答案带来权值；所以在模k的完全剩余系中对应元素连边（对于一个$x$，我们给$x+1\%k$和$10\ast 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]);
	}
}