PAT A1103

题目:https://pintia.cn/problem-sets/994805342720868352/problems/994805364711604224

我真的是蠢到爆了,我想着输出时需要从大到小,而大在前,小在后,就用队列来输出,于是把答案往队列queue中存。忽略了致命的错误,pop的时候,是最前面的被pop而不是选不选当前index。所以只能用vector。

#include <cstdio>
#include <cmath>
#include <queue>
using namespace std;
queue<int> temp, ans;
int n, k, p, maxsum;
void DFS(int index, int nowK, int x, int sum){
	if(index == 0 || x > n || nowK > k)
		return;
	if(nowK == k && x == n){
		if(ans.empty() || sum > maxsum){
			ans = temp;
			maxsum = sum;
		}
		return;
	}
	temp.push(index);
	DFS(index, nowK + 1, x + pow((double)index, (double)p), sum + index);
	temp.pop();
	DFS(index - 1, nowK, x , sum);
}
int main(){
	scanf("%d%d%d", &n, &k, &p);
	int j = pow((double)n, 1.0 / p);
	DFS( j, 0, 0, 0);
	if(!ans.empty()){
		printf("%d =", n);
		for(int i = 0; i < k; i++){
			printf(" %d^%d", ans.front(), p);
			ans.pop();
			if(i < k - 1)
				printf(" +");
		}
		printf("\n");
	}
	else
		printf("Impossible\n");
	return 0;
}

修正后:

#include <cstdio>
#include <cmath>
#include <vector>
using namespace std;
vector<int> temp, ans;
int n, k, p, maxsum;
void DFS(int index, int nowK, int x, int sum){
	if(index == 0 || x > n || nowK > k)
		return;
	if(nowK == k && x == n){
		if(ans.empty() || sum > maxsum){
			ans = temp;
			maxsum = sum;
		}
		return;
	}
	temp.push_back(index);
	DFS(index, nowK + 1, x + pow((double)index, (double)p), sum + index);
	temp.pop_back();
	DFS(index - 1, nowK, x , sum);
}
int main(){
	scanf("%d%d%d", &n, &k, &p);
	int j = pow((double)n, 1.0 / p);
	DFS( j, 0, 0, 0);
	if(!ans.empty()){
		printf("%d =", n);
		for(int i = 0; i < k; i++){
			printf(" %d^%d", ans[i], p);
			if(i < k - 1)
				printf(" +");
		}
		printf("\n");
	}
	else
		printf("Impossible\n");
	return 0;
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值