1103 Integer Factorization (30 分)dfs+剪枝

最新推荐文章于 2020-07-21 17:38:22 发布

青春苦瓜

最新推荐文章于 2020-07-21 17:38:22 发布

阅读量157

点赞数

分类专栏： PAT

本文链接：https://blog.csdn.net/qq_41214789/article/details/103083092

版权

PAT 专栏收录该内容

14 篇文章 0 订阅

订阅专栏

没啥好说的
感觉自己剪枝已经到极限了可还是T
后来想了想，我的垃圾思路是每次都得从头到尾遍历一遍
正解是从最大开始判断当前数字放还是不放，所以不用从头到尾遍历也不用判断sum一样时候字典序大小
T了的垃圾代码：

#include<iostream>
#include<algorithm>
#include<string.h>
#include<map>
#include<queue>
#include<cmath>
#include<cstdio>
#include<vector>
#define ll long long
#define inf 0x3f3f3f3f
using namespace std;
const int maxn=1e3+5;
int n,k,p;
int limit;
int lujing[405],flag=0;
int ans[405],anssum,poww[500];
bool cmp()
{
	for(int i=0;i<k;i++)
	{
		if(ans[i]<lujing[i])
			return 1;
		else if(ans[i]>lujing[i])
			return 0;
	}
	return 0;
}
void dfs(int now,int step)
{
	if(step>k)
		return;
	if(now==n&&step==k)
	{
		if(!flag)
		{
			for(int i=0;i<k;i++)
				ans[i]=lujing[i],anssum+=lujing[i];
			flag=1;
		}
		else
		{
			int sum=0;
			for(int i=0;i<step;i++)
				sum+=lujing[i];
			if(sum>anssum)
			{
				for(int i=0;i<step;i++)
					ans[i]=lujing[i];
				anssum=sum;
			}
			else if(sum==anssum&&cmp())
			{
				for(int i=0;i<step;i++)
					ans[i]=lujing[i];
				anssum=sum;
			}
		}
		return;
	}
	for(int i=1;i<limit;i++)
	{
		lujing[step]=i;
		if(now+poww[i]<=n&&step+1<=k)
			dfs(now+poww[i],step+1);
		else return;
	}
}
int main()
{
	scanf("%d%d%d",&n,&k,&p);
	limit=1;
	poww[limit]=1;
	while(poww[limit]<=n)
	{
		limit++;
		poww[limit]=pow(limit,p);
	}
	dfs(0,0);
	if(!flag)
		printf("Impossible");
	else
	{
		printf("%d = ",n);
		for(int i=0;i<k;i++)
		{
			printf("%d^%d",ans[i],p);
			if(i!=k-1)
				printf(" + ");
		}
	}
	return 0;
}

看了题解后改的代码

#include<iostream>
#include<algorithm>
#include<string.h>
#include<map>
#include<queue>
#include<cmath>
#include<cstdio>
#include<vector>
#define ll long long
#define inf 0x3f3f3f3f
using namespace std;
const int maxn=1e3+5;
int n,k,p;
vector<int> lujing,ans;
int flag=0,anssum,limit;
int poww[500];
void dfs(int up,int now,int step,int factsum)
{
	if(step>k||now>n)
		return;
	if(step==k&&now==n)
	{
		if(!flag)
		{
			ans=lujing;
			flag=1;
			anssum=factsum;
		}
		else
		{
			if(anssum<factsum)
			{
				anssum=factsum;
				ans=lujing;
			}
		}
	}
	lujing.push_back(up);
	dfs(up,now+poww[up],step+1,factsum+up);
	lujing.pop_back();
	if(up>1)
		dfs(up-1,now,step,factsum);
}
int main()
{
	scanf("%d%d%d",&n,&k,&p);
	limit=1;
	poww[limit]=1;
	while(poww[limit]<=n)
	{
		limit++;
		poww[limit]=pow(limit,p);
	}
	dfs(limit-1,0,0,0);
	if(!flag)
		printf("Impossible");
	else
	{
		printf("%d = ",n);
		for(int i=0;i<k;i++)
		{
			printf("%d^%d",ans[i],p);
			if(i!=k-1)
				printf(" + ");
		}
	}
	return 0;
}