P3011 [USACO11JAN] Traffic Lights S 题解

题目不难，跑个spfa就行。

但是在松弛操作的时候，有两种情况需要考虑：

• 当前时间，一条边连接的两座城市状态相同。
• 两座城市状态不同，需要等待状态统一。

第一种情况可以直接松弛并入队，第二种情况为了（懒）避免复杂的代码操作，所以我们可以通过一种朴实无华的简单思路来实现状态统一：我们可以将当前城市的时间加一，再次入队。

这样就可以搞定第二种情况了，但是代码跑的比较慢，需要加上一个优化：当时间过大时，不再继续通过该状态更新其他点。

剩下的就是判断当前时间当前点颜色了，自己思考吧hhh。

记得无解情况输出0。

贴上核心代码：

// 判断颜色
inline char getc(LL x,LL t){
    if(t<sy[x]){
        return c[x];
    }else{
        if(c[x]=='B'){
            if((t-sy[x])%(b[x]+p[x])-p[x]>=0){
                return 'B';
            }else{
                return 'P';
            }
        }else{
            if((t-sy[x])%(b[x]+p[x])-b[x]>=0){
                return 'P';
            }else{
                return 'B';
            }
        }
    }
}

void spfa(){
    memset(dis,0x3f,sizeof(dis));
    dis[s]=0;
    q.push(node{s,0});
    vis[s][0]=true;
    while(!q.empty()){
        node t=q.front();
        q.pop();
        if(t.t>100000)continue;//优化，避免超时
        vis[t.x][t.t]=false;
        for(LL i=0;i<a[t.x].size();i++){
            LL xx=a[t.x][i].first;
            LL yy=a[t.x][i].second;
            node x;
            if(t.t+yy<dis[xx]){
                if(getc(xx,t.t)==getc(t.x,t.t)){//如果状态相同
                    dis[xx]=t.t+yy;//就入队并更新
                    x.x=xx;
                    x.t=t.t+yy;
                }else{//如果不同
                    x.x=t.x;//就在当前城市基础上时间加一再次入队
                            //直到两个城市状态统一的时候，就会更新了
                    x.t=t.t+1;
                }
                if(!vis[x.x][x.t]){
                    vis[x.x][x.t]=true;
                    q.push(x);
                }
            }
        }
    }
}