1103 Integer Factorization (30 分)

1103 Integer Factorization (30 分)

题意

给出三个正整数，N，K , P 要求找到一组数使得 N = n[1]^P+n[2] ^P+…n[K] ^P，且n[1]+n[2]+…n[k]最大（有多组这样的数时最大的一组），有多组一样大时选字典序最大的一组。没有这样的组合时输出 “Impossible”

思路

可以考虑深度优先搜索找出所有组合，然后从中选出符合要求的组合。
考虑到 n[i]^p<=N,可以得到n[i]的范围。用数组把n[i]可能取得的数存起来，然后dfs找到这些数中符合要求的组合。
数据较大时dfs容意超时，需要优化时间复杂度。有这几点可以下手：
1.当选取的数之和大于N时后面无论怎么组合也得不到N；
2.当选取的数超过k个时组合不符合要求；
3.符合要求的组合是单调不增的序列。

代码

#include<stdio.h>
#include<stack>
#include<math.h>
#include<string.h>
#include<vector>
using namespace std;
//typedef struct node{
//	int sum,list[30],x;
//}node;
//node
int x[30],xmax=0,sum,num,vis[30];
int k,n,p,cnt,temp;
vector<int> list,y;
int dfs(int sum,int count,int num)
{
//	for(int i=0;i<count;i++){
//		printf("%d ",list[i]);
//	}
//	printf("###\n");
	if(count>=k||sum>n){
		if(sum>n){
			return 0;
		}
		if(sum==n&&num>xmax&&count==k){
			xmax=num;
//			printf("%d###\n",num);
//			for(int i=0;i<count;i++){
//				y[i]=list[i];
//			}
			y=list;
		}
		return 0;
	}
	for(int i=temp;i>0;i--){
//		if(vis[i]==0){
//			list[count]=i;
			list.push_back(i);
//			vis[i]=1;
			temp=i;
			dfs(sum+x[i],count+1,num+i);
//			vis[i]=0;
			list.pop_back();
//		}
	}
}
int main()
{
	memset(vis,0,sizeof(vis));
	scanf("%d%d%d",&n,&k,&p);
	for(int i=0;pow(i,p)<=n;i++){
		x[i]=pow(i,p);
		cnt=i;
	}
	cnt++;
	temp=cnt-1;
	dfs(0,0,0);
	if(xmax!=0){
		printf("%d = ",n);
		for(int i=0;i<k;i++){
			if(i!=0){
				printf(" + %d^%d",y[i],p);
			}else{
				printf("%d^%d",y[i],p);
			}
		}
		printf("\n");
	}else{
		printf("Impossible\n");
	}
	return 0;
}