luogu 3403跳楼机

题目描述

Srwudi的家是一幢h层的摩天大楼。由于前来学习的蒟蒻越来越多，srwudi改造了一个跳楼机，使得访客可以更方便的上楼。经过改造，srwudi的跳楼机可以采用以下四种方式移动：
向上移动x层；
向上移动y层；
向上移动z层；
回到第一层。
一个月黑风高的大中午，DJL来到了srwudi的家，现在他在srwudi家的第一层，碰巧跳楼机也在第一层。DJL想知道，他可以乘坐跳楼机前往的楼层数。
输入输出格式
输入格式：
第一行一个整数h，表示摩天大楼的层数。
第二行三个正整数，分别表示题目中的x, y, z。

输出格式：
一行一个整数，表示DJL可以到达的楼层数。
输入输出样例
输入样例#1：
15
4 7 9
输出样例#1：
9

输入样例#2：
33333333333
99005 99002 100000
输出样例#2：
33302114671
说明
可以到达的楼层有：1,5,8,9,10,12,13,14,15
1<=h<=2^63-1，1<=x, y, z<=100000

分析

一道经典的同余类BFS题，根据题意，可先将x，y，z排序，使得x＜y＜z，设f[i]表示到达余数为i的楼层的数，根据楼数%x的余数，有0,1，……x-1，将其与y，z连边，跑一边SPFA后可得出f[i]的值，后直接处理即可，时间：60ms。
上代码

#include<bits/stdc++.h>
using namespace std;
long long al[4],fi[100001],w[200001],ne[200001],v[200001],tot,h,f[100001],ans;
bool b[100001];
queue<int> q;
void add(int u,int vv,int val){
    w[++tot]=vv,ne[tot]=fi[u];
    fi[u]=tot,v[tot]=val;
}
void spfa(){
    memset(f,0x3f,sizeof(f));
    f[1]=1,b[1]=1,q.push(1);
    while(!q.empty()){
        int k=q.front();
        q.pop();
        for(int i=fi[k];i;i=ne[i])
            if(f[w[i]]>f[k]+v[i]){
                f[w[i]]=f[k]+v[i];
                if(!b[w[i]])
                    b[w[i]]=1,q.push(w[i]);
            }
        b[k]=0;
    }
}
int main(){
    scanf("%lld%d%d%d",&h,&al[1],&al[2],&al[3]);
    sort(al+1,al+4);
    for(int i=0;i<al[1];i++)
        add(i,(i+al[2])%al[1],al[2]),add(i,(i+al[3])%al[1],al[3]);
    spfa();
    for(int i=0;i<al[1];i++)
        if(h>=f[i])
            ans+=(h-f[i])/al[1]+1;
    printf("%lld\n",ans);
    return 0;
}