算法趣题-Q18

一、问题描述

 

二、问题分析

        做此题的时候陷入了思维误区，总想着如何用算法去直接得出答案，忘记了使用搜索的办法，当然直接暴力的话运行还是比较慢的，需要根据情况去减少遍历的次数。这里，个人使用的是按相邻两个数的和是平方数的前提下进行搜索，最后使用递归方法进行代码实现。

三、代码实现

1.C/C++实现

#include <iostream>
#include <vector>

using namespace std;

vector<int> used;
vector<int> squares = { 1, 4, 9 };

bool getNext(int n)
{
	if (used.size() == n && find(squares.begin(), squares.end(), (n + used[n - 1])) != squares.end())
		return true;
	for (unsigned int i = 1; i < squares.size(); i++)
	{
		int tmp = squares[i] - used.back();
		// 将 squares[i] - used.back()的值锁定在区间 (0, n], 且未被排列
		if (tmp > 0 && tmp <= n && find(used.begin(), used.end(), tmp) == used.end())
		{
			used.push_back(tmp);
			if (getNext(n))
				return true;
			used.pop_back();
		}
	}
	return false;
}

int main()
{
	int n = 1;
 	while (n++)
	{
		// 扩充平方数组长度
		while (squares.back() < n * 2)
			squares.push_back((squares.size() + 1) * (squares.size() + 1));
		used.push_back(n);
		if (getNext(n))
			break;
		used.pop_back();
	}
	cout << n << endl;
}

2.Python实现

# coding=utf-8

square_list = [1]


def get_next(cur, alter_list, used):
    # 递归出口，所有的数都用过了，验证首尾
    if len(used) == used[0]:
        return (n + cur) in square_list

    # 遍历可能的下一个
    for item in alter_list:
        if item in used or item > used[0]:
            continue
        used.append(item)
        if get_next(item, [square - item for square in square_list if square > item], used):
            return True
        used.remove(item)
    return False


if __name__ == '__main__':
    n = 1
    while True:
        n += 1
        # 扩展平方数列表
        while square_list[-1] < n * 2:
            square_list.append((len(square_list) + 1) ** 2)

        # 以 n 为起始进行DFS
        if get_next(n, [item - n for item in square_list if item > n], [n, ]):
            break
    print(n)