hdu1548A strange lift——最短路(迪杰斯特拉,spfa)。bfs(队列。数组)

7 篇文章 0 订阅
6 篇文章 0 订阅

题意:有n层电梯,当前处于A层,要到B层去,然后给你n个数,代表处在i层时会向下或者上走a[i]层,即到达i+a[i]或i-a【i】,求最小按电梯的次数(上下移动次数);

思路:简单题,没有什么思路,用最短路的Dijkstra和spfa就是简单地模板,两个bfs也挺简单。

我用四中方法做的,

首先是队列模拟bfs

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<queue>
using namespace std;
int a[250];//记录每层电梯上下的层数 
int n,s,e;//n层,s是起始层,e是终止层 
int mark[250];
int step[250];//记录到i层要step【i】步 
int loc[250];//没有队列,就用这个数组模拟队列,存放走过的 
void bfs()
{
	int length,tail;
	length=1;//增加和表示“队列”的长度 
	tail=0;//在数组里取出下一个数 
	loc[0]=s;
	while(length>tail)
	{
		int x=loc[tail++];
		if(x==e)
		{
			printf("%d\n",step[x]);
			return;
		}
		int up=x+a[x];
		if(mark[up]==0&&up>=1&&up<=n)
		{
			loc[length++]=up;
			mark[up]=1;
			step[up]=step[x]+1;
		}
		int down=x-a[x];
		if(mark[down]==0&&down>=1&&down<=n)
		{
			loc[length++]=down;
			mark[down]=1;
			step[down]=step[x]+1;
		}
	}
	printf("-1\n");
}
int main()
{
	
	while(scanf("%d",&n)!=EOF)
	{
		if(n<=0) break;
		scanf("%d%d",&s,&e);
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&a[i]);
			step[i]=0;
			mark[i]=0;
		}
		if(s==e)
		{
			printf("0\n");
			continue;
		}
		bfs();
	}
	return 0;
}

bfs队列做法:
#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<queue>
using namespace std;
int a[250];//记录每层电梯上下的层数 
int mark[250];
int step[250];//记录到i层要step【i】步 
int n,s,e;//n层,s是起始层,e是终止层 
void bfs()
{
	queue<int>q;
	int x,down,up;
	while(!q.empty()) q.pop();
	q.push(s);
	while(!q.empty())
	{
		x=q.front();
		q.pop();
		if(x==e)
		{
			printf("%d\n",step[x]);
			return;
		}
		up=x+a[x];
		if(up>=1&&up<=n&&mark[up]==0)
		{
			q.push(up);
			mark[up]=1;
			step[up]=step[x]+1;
		}
		down=x-a[x];
		if(down>=1&&down<=n&&mark[down]==0)
		{
			q.push(down);
			mark[down]=1;
			step[down]=step[x]+1;
		}
	}
	printf("-1\n");
}
int main()
{
	while(scanf("%d",&n)!=EOF)
	{
		if(n<=0) break;
		scanf("%d%d",&s,&e);
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&a[i]);
			step[i]=0;
			mark[i]=0;
		}
		if(s==e)
		{
			printf("0\n");
			continue;
		}
		bfs();
		
	}
	return 0;
}
迪杰斯特拉做法:

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#define inf 5000;
using namespace std;
int n,s,e;
int a[250];
int dis[250];
int map[250][250];
int mark[250];
void Dijkstra()
{
	for(int i=1;i<=n;i++)
	{
		dis[i]=map[s][i];
		mark[i]=0;
	}
	for(int i=1;i<=n;i++)
	{
		int x,m=inf;
		for(int j=1;j<=n;j++)
		{
			if(dis[j]<m&&mark[j]==0)
			{
				m=dis[j];
				x=j;
			}
		}
		mark[x]=1;
		for(int j=1;j<=n;j++)
		{
			if(dis[x]+map[x][j]<dis[j]&&mark[j]==0)
			{
				dis[j]=dis[x]+map[x][j];
			}
		}
	}
}
int main()
{
	while(scanf("%d",&n)!=EOF)
	{
		if(n<=0) break;
		scanf("%d%d",&s,&e);
		for(int i=1;i<=n;i++)
		{
			for(int j=1;j<=n;j++)
			{
				map[i][j]=inf;
			}
		}
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&a[i]);
			if(i+a[i]>=1&&i+a[i]<=n)
			{
				map[i][i+a[i]]=1;
			}
			if(i-a[i]>=1&&i-a[i]<=n)
			{
				map[i][i-a[i]]=1;
			}
		}
		if(s==e)
		{
			printf("0\n");
			continue;
		}
		Dijkstra();
		if(dis[e]==5000)
		{
			printf("-1\n");
		}
		else
		{
			printf("%d\n",dis[e]);
		}
	}
	return 0;
}

spfa做法:

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<queue>
#define inf 50000
using namespace std;
struct node
{
	int next;
	int to;
	int weight;
}life[550];
int mark[250];
int dis[250];
int head[250];
int a[250];
int n,s,e;int cur;
void init()
{
	cur=1;
	memset(head,-1,sizeof(head));
	memset(life,0,sizeof(life));
	for(int i=1;i<=n;i++)
	{
		dis[i]=inf;
		mark[i]=0;
	}
}
void add(int u,int v)
{
	life[cur].to=v;
	life[cur].weight=1;
	life[cur].next=head[u];
	head[u]=cur;
	cur++;
}
void spfa_bfs()
{
	queue<int>q;
	int first,i;
	while(!q.empty()) q.pop();
	q.push(s);
	dis[s]=0;
	mark[s]=1;
	while(!q.empty())
	{
		first=q.front();
		q.pop();
		mark[first]=0;
		for(i=head[first];i!=-1;i=life[i].next)
		{
			if(life[i].weight+dis[first]<dis[life[i].to])
			{
				dis[life[i].to]=life[i].weight+dis[first];
				if(mark[life[i].to]==0)
				{
					q.push(life[i].to);
					mark[life[i].to];
				}
			}
		}
	}
}
int main()
{
	while(scanf("%d",&n)!=EOF)
	{
		if(n<=0) break;
		init();
		scanf("%d%d",&s,&e);
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&a[i]);
			if(i+a[i]<=n)
			add(i,i+a[i]);
			if(i-a[i]>=1)
			add(i,i-a[i]);
		}
		if(s==e)
		{
			printf("0\n");
			continue;
		}
		spfa_bfs();
		if(dis[e]>=inf)
		{
			printf("-1\n");
		}
		else
		{
			printf("%d\n",dis[e]);
		}
	}
	return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
对于HDU4546问题,还可以使用优先队列(Priority Queue)来解决。以下是使用优先队列的解法思路: 1. 首先,将数组a进行排序,以便后续处理。 2. 创建一个优先队列(最小堆),用于存储组合之和的候选值。 3. 初始化优先队列,将初始情况(即前0个数的组合之和)加入队列。 4. 开始从1到n遍历数组a的元素,对于每个元素a[i],将当前队列中的所有候选值取出,分别加上a[i],然后再将加和的结果作为新的候选值加入队列。 5. 重复步骤4直到遍历完所有元素。 6. 当队列的大小超过k时,将队列中的最小值弹出。 7. 最后,队列中的所有候选值之和即为前k小的组合之和。 以下是使用优先队列解决HDU4546问题的代码示例: ```cpp #include <iostream> #include <vector> #include <queue> #include <functional> using namespace std; int main() { int n, k; cin >> n >> k; vector<int> a(n); for (int i = 0; i < n; i++) { cin >> a[i]; } sort(a.begin(), a.end()); // 对数组a进行排序 priority_queue<long long, vector<long long>, greater<long long>> pq; // 最小堆 pq.push(0); // 初始情况,前0个数的组合之和为0 for (int i = 0; i < n; i++) { long long num = pq.top(); // 取出当前队列中的最小值 pq.pop(); for (int j = i + 1; j <= n; j++) { pq.push(num + a[i]); // 将所有加和结果作为新的候选值加入队列 num += a[i]; } if (pq.size() > k) { pq.pop(); // 当队列大小超过k时,弹出最小值 } } long long sum = 0; while (!pq.empty()) { sum += pq.top(); // 求队列中所有候选值之和 pq.pop(); } cout << sum << endl; return 0; } ``` 使用优先队列的方法可以有效地找到前k小的组合之和,时间复杂度为O(nklog(k))。希望这个解法对你有所帮助!

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值