PAT 1104 Sum of Number Segments （暴力DFS）

最新推荐文章于 2020-12-16 16:23:08 发布

h1021456873

最新推荐文章于 2020-12-16 16:23:08 发布

阅读量293

点赞数

分类专栏： PAT甲级 dfs

本文链接：https://blog.csdn.net/h1021456873/article/details/53538629

版权

dfs 同时被 2 个专栏收录

37 篇文章 0 订阅

订阅专栏

PAT甲级

18 篇文章 0 订阅

订阅专栏

Given a sequence of positive numbers, a segment is defined to be a consecutive subsequence. For example, given the sequence {0.1, 0.2, 0.3, 0.4}, we have 10 segments: (0.1) (0.1, 0.2) (0.1, 0.2, 0.3) (0.1, 0.2, 0.3, 0.4) (0.2) (0.2, 0.3) (0.2, 0.3, 0.4) (0.3) (0.3, 0.4) (0.4).

Now given a sequence, you are supposed to find the sum of all the numbers in all the segments. For the previous example, the sum of all the 10 segments is 0.1 + 0.3 + 0.6 + 1.0 + 0.2 + 0.5 + 0.9 + 0.3 + 0.7 + 0.4 = 5.0.

Input Specification:

Each input file contains one test case. For each case, the first line gives a positive integer N, the size of the sequence which is no more than 10⁵. The next line contains N positive numbers in the sequence, each no more than 1.0, separated by a space.

Output Specification:

For each test case, print in one line the sum of all the numbers in all the segments, accurate up to 2 decimal places.

Sample Input:

4
0.1 0.2 0.3 0.4

Sample Output:

5.00

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
int a[22],N,K,P,b[1111],vis[1111],flag;

void dfs(int dis,int sum,int cnt)
{
	int i;
	if(sum>N || cnt>K) return;
	if(sum==N && cnt==K) {
		for(i=1;i<=20;i++) b[i]=vis[i];
		flag=1;
		return;
	}
	for(i=dis;i<=20;i++) {
		if(sum+a[i]>N) break;
		vis[i]++;
		dfs(i,sum+a[i],cnt+1);
		vis[i]--;
	}
}

int main()
{
	int i,j,f;
	scanf("%d%d%d",&N,&K,&P);
	for(i=1;i<=20;i++) {
		a[i]=i;
		for(j=1;j<P;j++)
		a[i]*=i;
	}
	flag=0;
	dfs(1,0,0);
	if(!flag) printf("Impossible\n");
	else {
		f=0;
		printf("%d =",N);
		for(i=20;i;i--) {
			while(b[i]--) {
				if(f) printf(" +");
				f=1;
				printf(" %d^%d",i,P);
			} 
		}
		printf("\n");
	} 
	return 0;
}