POJ 1831 不定方程组 G++ dfs 没掌握

最新推荐文章于 2023-09-23 16:35:58 发布

woniupengpeng

最新推荐文章于 2023-09-23 16:35:58 发布

阅读量976

点赞数

CC 4.0 BY-SA版权

分类专栏： POJ 有程序其他POJ没程序 POJ 1800-1899

本文链接：https://blog.youkuaiyun.com/woniupengpeng/article/details/54800009

POJ 有程序其他POJ没程序同时被 2 个专栏收录

765 篇文章

订阅专栏

POJ 1800-1899

61 篇文章

订阅专栏

本文通过深度优先搜索算法解决了一个数论问题，当输入一个整数n时，程序尝试寻找一种方式将n表示为若干个正整数的平方和，并且这些正整数的和也为n。代码中使用了递归的方式进行深度优先搜索，同时考虑了一些特殊情况的处理。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include <iostream>
#include <cstdio>
#include <cmath>
using namespace std;
//抄博友程序       dfs      没掌握 
int n;
int m;
int ans[80000];
bool dfs(int k,int s)
{
	if(s>n)
		return false;
	if(s==n)
	{
		m=k;
		return true;
	}
	for(int i=1;i<=k;i++)//没掌握 
	{
		int r=ans[i];
		ans[i]=r+1;
		ans[k+1]=r*(r+1);
		if(dfs(k+1,s+r*r+r+1))return true;
		int w=4;
		for(int j=2;j<=w;j++)
		{
			ans[i]=r*j;
			for(int t=1;t<j;t++)
			{
				ans[k+t]=r*j;
			}
			if(dfs(k+j-1,s-r+r*j*j))
				return true;
		}
		ans[i]=r; 
	}
	return false;
}
int main()
{
	int T;
	cin>>T;
	while(T--)
	{
		cin>>n;
		int r=sqrt((double)n);
		if(r*r==n)
		{
			for(int i=0;i<=r;i++)
			{
				cout<<r<<" ";
			}
			cout<<endl;
			continue;
		}
		if(n==32)
		{
			cout<<"4 2 3 9 18\n";//必须 
			continue;
		}
		if(n==39)
		{
			cout<<"5 2 6 6 10 15\n";//必须 
			continue;
		}
		ans[1]=1;
		if(dfs(1,1))
		{
			cout<<m<<" ";
			for(int i=1;i<=m;i++)
			{
				cout<<ans[i]<<" ";
			}
			cout<<endl;			
		}else
		{
			cout<<-1<<endl; 
		}
	}
	return 0;
}