标签: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;
}
标签:pop main nbsp def div 方案 highlight 方法 code
原文地址:https://www.cnblogs.com/asdfknjhu/p/12442003.html