当前位置: 代码迷 >> 综合 >> POJ 1564 Sum It Up(DFS)
  详细解决方案

POJ 1564 Sum It Up(DFS)

热度:48   发布时间:2023-12-08 11:18:21.0

题目链接:
POJ 1564 Sum It Up
题意:
给你一个非递增数组,输出由数组中的数字组合的相加和等于给定t的所有可能组合。
每个组合内部按照非递增顺序输出,每个组合按照首数字从大到小排序,首数字相同比较第二个数字,依次类推。
分析:
输出要求按照DFS的搜索顺序存储输出即可。
DFS过程中需要记录有哪些数字被使用了,以及当前和,当前遍历的数字编号。
注意判重。

CODE:

//816K 0MS
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <cstring>
using namespace std;const int maxn=15;int t,n,tot;
int ans[30000][maxn],a[maxn],vis[maxn],b[maxn],cnt[30000];void dfs(int k,int sum)//计算到下标为k,包括a[k]在内的和为sum
{if(sum==t){int tmp=0;for(int i=0;i<=k;i++){if(vis[i])b[tmp++]=a[i];}//记录当前和的组成int ok=0;for(int i=0;i<tot;i++){if(cnt[i]==tmp){int flag=0;for(int j=0;j<tmp;j++){if(ans[i][j]!=b[j]){flag=1;break;}}if(flag==0) ok=1;//完全相同if(ok) break;}}if(ok) return;//重复了for(int i=0;i<tmp;i++)ans[tot][i]=b[i];cnt[tot]=tmp;tot++;return;}if(k==n) return;for(int i=k+1;i<n;i++){vis[i]=1;dfs(i,sum+a[i]);vis[i]=0;}
}int main()
{
#ifdef LOCALfreopen("in.txt","r",stdin);
#endifwhile(~scanf("%d%d",&t,&n)){if(t==0&&n==0) break;for(int i=0;i<n;i++)scanf("%d",&a[i]);memset(cnt,0,sizeof(cnt));tot=0;for(int i=0;i<n;i++){memset(vis,0,sizeof(vis));vis[i]=1;dfs(i,a[i]);}printf("Sums of %d:\n",t);if(tot==0) printf("NONE\n");else{for(int i=0;i<tot;i++){for(int j=0;j<cnt[i];j++){if(j) printf("+");printf("%d",ans[i][j]);}printf("\n");}}}return 0;
}