UVA11111- Generalized Matrioshkas

题意:嵌套娃娃,在里层的娃娃的尺寸要小于外层的娃娃,有可能一个娃娃里面有多个娃娃,而这多个娃娃尺寸和要小于外层的,相当与匹配括号的加强版,有些但疼。

思路:要多容量大小是否足够的判断,小看了别人的一些思路,自己写了代码。下面的核心代码是deal函数,用的是错位的方法(个人觉得),同一top的sum[top]和stack[top],
其实是sum[top]是在stack[top]内层的容量,比如-5 -3 3 5,sum[0] = 5,stack[1] = -5,sum[1] = 3,stack[2] = -3。这样就可以判断是否有足够的容量装多个娃娃。
(感觉还是有点抽象的样子,自己在草稿纸上写写就会明白了。。。)


#include<stdio.h> 
#include<string.h>
#include<math.h>
#include<algorithm>
using namespace std;
#define N 1000000

int arr[N], stack[N], sum[N];

int deal(int *a, int n) {
	int top = 0;
	memset(stack, 0, sizeof(stack));
	memset(sum, 0, sizeof(sum));哦
	for(int i = 0; i < n; i++){
		if(a[i] < 0) {  
			sum[top] -= a[i];//记录每一层的容量  
		}  
		else {  
			if(-stack[top] != a[i])//判断正数是否与栈顶的元素和为0
				return 0;   
			if(sum[top] >= a[i])//每一个进行判断的正数,要求要比内层的容量大
				return 0;  
			stack[top] = 0;//元素出栈
			sum[top--] = 0;//当元素出栈时,其相应的容量要清零
		}
	}
	return 1;	
}

int main() {
	int cnt = 0;
	char ch;
	while (scanf("%d", &arr[cnt++]) != EOF) {
		int flag = 0;
		ch = getchar();		
		if (ch == '\n') {
			if (cnt % 2 == 1)
				flag = 0;
			else	
				flag = deal(arr, cnt);	
			if (flag)			
				printf(":-) Matrioshka!\n");
			else
				printf(":-( Try again.\n");
			cnt = 0;
		}
	}
	return 0;
}


你可能感兴趣的:(UVA11111- Generalized Matrioshkas)