Codeforces Round #637 (Div. 2) - Thanks, Ivan Belonogov! E—Nastya and Unexpected Guest 01bfs dp

dp[i][j],到达i时绿灯已经亮了j秒,经过的最少轮数(一轮表示绿灯和红灯各亮一遍)

显然每一步可以向左或向右走,写一个01bfs+dp转移即可。

转移时,判是否可以往前走或往后走一步。

要么是绿灯期间过去,轮数不变。

要么过去时刚好变成红灯,轮数加1.

具体看代码。

复杂度O(gm),即01BFS的复杂度。注意记得记忆化,即如果目的地相同状态的轮数更优,则没必要进行转移。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define ls (o<<1)
#define rs (o<<1|1)
#define pb push_back
const double PI= acos(-1.0);
const int M = 1e5+7;
/*
int head[M],cnt;
void init(){cnt=0,memset(head,0,sizeof(head));}
struct EDGE{int to,nxt,val;}ee[M*2];
void add(int x,int y){ee[++cnt].nxt=head[x],ee[cnt].to=y,head[x]=cnt;}
*/

ll dp[10100][1010],a[10100]; 
ll n,m,g,r;
struct node{
	int id,val;
};
deque<node>q;
ll bfs()
{
	memset(dp,0x3f,sizeof(dp));
	q.push_front(node{1,0});
	dp[1][0]=0;
	ll ans=1e18;
	while(q.size())
	{
		node tp=q.front();q.pop_front();
		
		if(n-a[tp.id]<=g-tp.val)//能直接走到终点了
		{
			ans=min(ans,dp[tp.id][tp.val]*(g+r)+tp.val+n-a[tp.id]);
		//	cout<<tp.id<<" "<<tp.val<<" "<<dp[tp.id][tp.val]<<"  "<<n-a[tp.id]<<"   "<<g-tp.val<<"=======  "<<endl;
		}
		if(tp.id<m)//可以向右走 
		{
			int cz=a[tp.id+1]-a[tp.id];
			int i=tp.id+1,j=cz+tp.val;
			if(j<g)//在绿灯时到达i 
			{
				if(dp[i][j]>dp[i-1][tp.val])
					dp[i][j]=dp[i-1][tp.val],q.push_front(node{i,cz+tp.val});//轮数不增加,边权为0,优先走
			}
			else if(cz+tp.val==g)//刚好在将要切红灯时到达i 
			{
				if(dp[i][0]>dp[i-1][tp.val]+1)
					dp[i][0]=dp[i-1][tp.val]+1,q.push_back(node{i,0});
			}
		}
		if(tp.id>1)//可以往左走
		{
			int cz=a[tp.id]-a[tp.id-1];
			int i=tp.id-1,j=cz+tp.val;
			if(j<g)//在绿灯时到达i 
			{
				if(dp[i][j]>dp[i+1][tp.val])
					dp[i][j]=dp[i+1][tp.val],q.push_front(node{i,cz+tp.val});//轮数不增加,边权为0,优先走
			}
			else if(cz+tp.val==g)//刚好在将要切红灯时到达i 
			{
				if(dp[i][0]>dp[i+1][tp.val]+1)
					dp[i][0]=dp[i+1][tp.val]+1,q.push_back(node{i,0});
			}
		} 
	}
	if(ans==1e18)return -1;
	return ans;
}
int main()
{
	ios::sync_with_stdio(false);
  	cin.tie(0);
  	cin>>n>>m;
  	for(int i=1;i<=m;i++)cin>>a[i];
  	sort(a+1,a+1+m);
  	cin>>g>>r;
  	ll ans=bfs();
  	cout<<ans<<endl;
	return 0;
}

 

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值