P4009 汽车加油行驶问题（spfa）

Posted 2021-07-24 SSL_LKJ

汽车加油

题目传送门

解题思路

建边
再spfa

AC代码

#include<cstring>
#include<cstdio>
#include<queue>
using namespace std;
int N,K,A,B,C,tot,c[1000005],head[1000005],a[105][105];
int dx[4]={-1,1,0,0};
int dy[4]={0,0,-1,1};
int dd[4]={1,0,1,0};
long long ans,d[1000005];
queue<int>q;
struct node
{
	int to,w,next;	
}aa[2000005];
int num(int x,int y,int z)//给个编号
{
	return z*N*N+(x-1)*N+y;
}
bool check(int x,int y)//判断
{
	if(x>=1&&x<=N&&y>=1&&y<=N)return 1;
	return 0;
}
void add(int x,int y,int z)//建表
{
	aa[++tot]=(node){y,z,head[x]};
	head[x]=tot;
}
void spfa()//spfa
{
	memset(d,0x7f,sizeof(d));
	c[num(1,1,K)]=1;
	d[num(1,1,K)]=0;
	q.push(num(1,1,K));
	while(!q.empty())
	{
		int x=q.front();
		q.pop();
		for(int i=head[x];i;i=aa[i].next)
		{
			int y=aa[i].to;
			if(d[y]>d[x]+aa[i].w*1ll)
			{
				d[y]=d[x]+aa[i].w*1ll;
				if(!c[y])c[y]=1,q.push(y);
			}
		}
		c[x]=0;
	}
}
int main()
{
	scanf("%d%d%d%d%d",&N,&K,&A,&B,&C);
	for(int i=1;i<=N;i++)
	 for(int j=1;j<=N;j++)
	  scanf("%d",&a[i][j]);
	for(int i=1;i<=N;i++)
	 for(int j=1;j<=N;j++)
	 {
	 	for(int k=0;k<K;k++)
	 	 if(a[i][j])add(num(i,j,k),num(i,j,K),A);
	 	 else add(num(i,j,k),num(i,j,K),A+C);
	 	for(int k=0;k<4;k++)
	 	{
	 		int x=i+dx[k],y=j+dy[k];
	 		if(check(x,y))
	 		{
	 			if(!a[i][j])
	 			 for(int kk=1;kk<=K;kk++)
	 			  if(dd[k])add(num(i,j,kk),num(x,y,kk-1),B);
	 			  else add(num(i,j,kk),num(x,y,kk-1),0);
	 			else 
	 			{
	 				if(dd[k])add(num(i,j,K),num(x,y,K-1),B);
	 				else add(num(i,j,K),num(x,y,K-1),0);
				}
			}
		}
	 }
	spfa();
	ans=d[0];
	for(int i=0;i<=K;i++)
	 ans=min(ans,d[num(N,N,i)]);
	printf("%lld",ans); 
	return 0;
}

谢谢