跳樓機[DP+spfa]
阿新 • • 發佈:2018-11-24
我們考慮對於一些樓層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; }