题目链接:点击进入
题目
思路
同余最短路是一种优化最短路建图的方法,通常解决给定 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;
}