P3403 跳楼机-（ 同余最短路 ）

题目链接：点击进入

题目

思路

同余最短路是一种优化最短路建图的方法，通常解决给定 n 个整数，问这 n 个整数能组成出多少个小于指定数字 h 的不同整数的问题 ( 这 n 个整数可以重复取 ) 或 给定 n 个整数，求这 n 个整数不能组成的最小（ 或最大 ）的整数的问题。
n 个整数 a1 - an ，我们选择一个最小的 ai ( a1 ) 作为 base ，然后把其他的 ai 表示成 base ∗ k + r 的形式。
余数 r 取值 [ 0 , base - 1 ] ，对于每个 r 建点，同时对于每个点连一条到点（ r + a [ i ] ）% base 的边，边长为 a [ i ]
然后从 0 开始跑最短路。
最后，枚举余数 r ，余数 r 的最短路为 dis [ r ] ，如果 dis [ r ] <= h 则在小于 h 的数中可以得到（ h − dis [ r ] ）/ a1 + 1 个整数，同时计入答案.（ dis [ i ] 记录凑出 mod base 余 i 最小的数是多少。）
这个题最后 h 要减一 , 因为是从一楼开始的，组成的数字范围最大为 h - 1

代码

#include<iostream>
#include<string>
#include<map>
#include<set>
#include<stack>
//#include<unordered_map>
#include<queue>
#include<cstdio>
#include<vector>
#include<cstring>
#include<algorithm>
#include<iomanip>
#include<cmath>
#include<fstream>
#define X first
#define Y second
#define best 131 
#define INF 0x3f3f3f3f3f3f3f3f
#define pii pair<int,int>
#define lowbit(x) x & -x
#define inf 0x3f3f3f3f
#define int long long
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const double pai=acos(-1.0);
const int N=2e4+10;
const int mod=1e9+7;
const double eps=1e-12;
const int maxn=1e6+10;
int n,h,a[maxn],head[maxn],tot,dis[maxn],vis[maxn];
struct node
{
	int val;
    int to;
	int next;
}edge[maxn*20];
void add(int u,int v,int w) 
{
	edge[tot].val=w;
	edge[tot].to=v;
	edge[tot].next=head[u];
	head[u]=tot++;
}
void spfa() 
{
    memset(dis,0x3f,sizeof(dis));
    queue<int>q;dis[0]=0;vis[0]=1;
    q.push(0);
    while(q.size()) 
	{
        int cur=q.front();q.pop();
        for(int i=head[cur];i!=-1;i=edge[i].next) 
		{
            int v=edge[i].to;
            if(dis[v]>dis[cur]+edge[i].val) 
			{
                dis[v]=dis[cur]+edge[i].val;
                if(!vis[v]) 
				{
                    vis[v]=1;
                    q.push(v);
                }
            }
        }
        vis[cur]=0;
    }
}
int solve(int x) 
{
    int res=0;
    for(int i=0;i<a[1];i++)
        if(dis[i]<=x) 
            res+=(x-dis[i])/a[1]+1;
    return res;
}
signed main() 
{
	memset(head,-1,sizeof(head));
    n=3;cin>>h;
    for(int i=1;i<=n;i++) cin>>a[i];
    for(int i=0;i<a[1];i++)
        for(int j=2;j<=n;j++)
            add(i,(i+a[j])%a[1],a[j]);
    spfa();
    cout<<solve(h-1);
    return 0;
}