uva11111 Generalized Matrioshkas (模拟 + 栈)-优快云博客

本文链接：https://blog.youkuaiyun.com/MistKafka/article/details/9471279

本文介绍了一种类似于括号匹配的算法，该算法确保每个数字只能与其相反数匹配，并且所有直接子括号的数值之和必须小于父括号的数值。通过使用栈来模拟这一过程，文章详细阐述了实现这一算法的具体步骤。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题意：类似于括号匹配，不过每个数字只能跟自己的相反数匹配。并且，括号的所有直接子括号的数值之和必须小于它的数值。

思路：用栈模拟，遇到负数入栈；正数则与栈顶匹配，若匹配成功则消去栈顶，否则也入栈。栈的元素是结构体，除了上面的括号自身的数值value外，还有所有直接子括号的和sum。在负数入栈前，要把它的数值加到栈顶（如果empty就不用）的sum，并判断，判断完才能入栈。

算法复杂度：n个括号操作n次。所以是o(N)。

代码：

#include <cstdio>
#include <stack>
using namespace std;

struct Num
{
	int value;
	int childSum;
};

stack<Num> sta;
bool ok;
int value;
char ch;

void init();
void doOnce();
bool judge();

int main()
{

	init();
	while (scanf("%d%c", &value, &ch) == 2) {
		if (ch == '\n')	 {
			doOnce();
			if (judge()) {
				printf(":-) Matrioshka!\n");
			} else {
				printf(":-( Try again.\n");
			}
			init();
		} else {
			doOnce();
		}
	}

	return 0;
}

void init()
{
	while (!sta.empty()) {
		sta.pop();
	}
	ok = true;
}

void doOnce()
{
	Num tp;
	tp.value = value;
	tp.childSum = 0;

	if (value < 0) {
		if (sta.empty()) {
			sta.push(tp);
		} else {
			sta.top().childSum += tp.value;
			if (sta.top().childSum <= sta.top().value) {
				ok = false;
			}
			sta.push(tp);
		}
	} else {
		if (sta.empty()) {
			ok = false;
		} else if (sta.top().value == -tp.value) {
			sta.pop();
		} else {
			sta.push(tp);
		}
	}
}

bool judge()
{
	if (ok && sta.empty()) {
		return true;
	} else {
		return false;
	}
}