1. 程式人生 > >跳樓機[DP+spfa]

跳樓機[DP+spfa]

傳送門

我們考慮對於一些樓層i, 當i可以被y或z走到時 , i+kx也可以被走到 , 對答案的貢獻就是(h-i)/x+1

於是我們預處理y或z走到modx==i的最少樓層

f[(i+y)%x] = min( f[(i+y)%x] , f[i%x] + y)   ,   z同理

發現對於狀態轉移 , i%x向(i+y)%x 連一條權值為y的邊 , 跑最短路就好了


#include<bits/stdc++.h>
#define N 500050
#define LL long long
using namespace std;
int first[N],next[N*2],to[N*2],w[N*2],tot;
LL h,x,y,z,dis[N],ans,vis[N];
void add(int x,int y,int z){
	next[++tot]=first[x],first[x]=tot,to[tot]=y,w[tot]=z;
}
void spfa(){
	queue<int> q; 
	memset(dis,127,sizeof(dis));
	dis[1]=1 , vis[1]=1 , q.push(1);
	while(!q.empty()){
		int x=q.front(); q.pop(); vis[x]=0;
		for(int i=first[x];i;i=next[i]){
			int t=to[i]; 
			if(dis[t] > (LL)dis[x] + w[i]){
				dis[t] = (LL)dis[x] + w[i];
				if(!vis[t]) q.push(t) , vis[t]=1;
			}
		}
	}
}
int main(){
	scanf("%lld%lld%lld%lld",&h,&x,&y,&z);
	for(int i=0;i<x;i++){
		add(i,(i+y)%x,y);
		add(i,(i+z)%x,z);
	} spfa();
	for(int i=0;i<x;i++) if(dis[i]<=h) ans += (LL)(h-dis[i])/x + 1ll; 
	printf("%lld",ans); return 0;
}