编程题-笔试-小米-数字串插入符号判断表达式结果

中午小米笔试的第二题编程题。

 

输入：整数N(1<=N<=7)，整数M；

你可以在数字串“123...N”中的两个字符间任意插入'+'或'-'，并计算插入后的字符串表达式结果；

输出：计算结果等于M的插入方式数；

 

例如输入N=3，M=2时，共有：

“123”=123，“12+3”=15，“12-3”=9，

“1+23”=24，“1+2+3”=6，“1+2-3”=0，

“1-23”=-22，“1-2+3”=2，“1-2-3”=-4这9种插入方式，其中只有1种计算结果为2，因此输出1。

 

思路：首先列举所有可能的插入方式，逐个计算表达式结果；计算时使用计算栈和符号栈（本题只用1个字符），扫描字符串遇到符号或结束时将数压栈，并进行一次二元数计算，计算数弹出栈、计算结果压栈，符号更新。

测试用例：

输入：

7 0

输出：

6

#include<iostream>;
#include<string>
#include<vector>
#include<stack>
using namespace std;

int M, N;
vector<string> v;	//存储所有可能的字符串
stack<int> t;	//计算栈

//列举所有插入方式字符串
void createString(string s, int n)
{
	char tmp = '0' + n;
	if (n==N)
	{
		v.push_back(s + tmp);
		return;
	}
	createString(s + tmp, n + 1);
	createString(s + tmp + '+', n + 1);
	createString(s + tmp + '-', n + 1);
}

//计算栈计算
void stackCal(char c)
{
	if (t.size()>1)
	{
		int t1 = t.top(), t2;
		t.pop();
		if (c == '+')
		{
			t2 = t1 + t.top();
		}
		else if (c == '-')
		{
			t2 = t.top() - t1;
		}
		t.pop();
		t.push(t2);
	}
}

//计算字符串表达式
int cal(string s)
{
	int len = s.length();
	char c=0;	//符号栈（这里其实只用一个符号因此一个字符就够)
	int tmp = 0;	//数暂存
	for (int i = 0; i < len; i++)
	{
		if (s[i]=='+')
		{
			t.push(tmp);
			tmp = 0;
			stackCal(c);
			c = '+';
		}
		else if (s[i] == '-')
		{
			t.push(tmp);
			tmp = 0;
			stackCal(c);
			c = '-';
		}
		else
		{
			tmp = tmp * 10 + s[i] - '0';
		}
	}
	t.push(tmp);
	stackCal(c);
	return t.top();
}

int main()
{
	while (cin>>N>>M)
	{
		v.clear();
		createString("", 1);

		int num = v.size(), count = 0;
		for (int i = 0; i < num; i++)
		{
			while (!t.empty())
			{
				t.pop();
			}
			int ans = cal(v[i]);
			if (ans==M)
			{
				count++;
			}
		}
		cout << count << endl;
	}
	return 0;
}