码迷,mamicode.com
首页 > 其他好文 > 详细

BFS — 最小方案数

时间:2020-03-08 14:28:40      阅读:61      评论:0      收藏:0      [点我收藏+]

标签:pop   main   nbsp   def   div   方案   highlight   方法   code   

https://www.bilibili.com/video/av32546525 代码与思路来自此

 

题目是:求 n 最少能表示成几个平方数的和

由于有 1*1=1 这个数,所以问题一定有解,因为 1 可以组成所有正整数

 

其中视频用到了宽搜,我是一点也没想到这种方法。

①  把 0 作为起始点,0+一个平方数 作为下一层,0+一个平方数+一个平方数  作为再下一层,由此可组成一个树

 

②  然后标记距离  0 距离为 0,0+一个平方数 距离为 1,0+一个平方数+一个平方数 距离为 2

 

③  所以 队列里面放的数为  0,0+一个平方数,0+一个平方数+一个平方数,这样循环结束的条件是 队列里 最先出现的那个 n

所以可以 是 那个取队首的变量 vertex == n

 

下面是我画的树:

技术图片

 

 下面是代码:

#define _CRT_SECURE_NO_WARNINGS
#include<stdio.h>
#include<stdlib.h>
#include<vector>
#include<queue>
using namespace std;
int BFS(int n)
{
	queue<int>q;
	vector<int>dist(n + 1, INT_MAX);   // 考虑极端条件 平方数全部由 1 组成,且还需要储存下一层 所以需要 n+1
	q.push(0);
	dist[0] = 0;

	while (q.size() > 0)
	{
		int vertex = q.front();    // 结点
		q.pop();

		if (vertex == n)
			return dist[vertex];
		for (int i = 1; i*i + vertex <= n; i++)
		{
			int next = i*i + vertex;          // 下一个结点
			if (dist[next] > dist[vertex] + 1)   // dist[next] 距离已经被标记好了,就是 dist[next] == dist[i] + 1 ,这就不用再放进队列里面了,因为这个数一定不会比之更小
			{
				q.push(next);
				dist[next] = dist[vertex] + 1;
			}
		}
	}

}
int main(void)
{
	int n;
	while (scanf("%d", &n) != EOF)
	{
		printf("%d\n", BFS(n));
	}
	system("pause");
	return 0;
}

  

 

BFS — 最小方案数

标签:pop   main   nbsp   def   div   方案   highlight   方法   code   

原文地址:https://www.cnblogs.com/asdfknjhu/p/12442003.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!