CSU 1378 Shipura 简单模拟

上周末中南的题，当时就知道是个简单模拟题，可是一个多小时就是没写出来，代码能力啊 >.<

题意：

　　某人发明了一种程序语言，只支持一种运算">>"，和一种函数"S<>"。现在给你一个表达式，让你求值。 表达式有以下几种元素：

expr : term 或者 expr+sp+">>"+sp+term

term : number 或者 "S"+sp+"<"+sp+expr+sp+">"

sp : "" 或者 " "

number : digit 或者 number+digeit

digit : "0" 或者 "1" 或者 "2" 或者 "3" 或者 "4" 或者 "5" 或者 "6" 或者 "7" 或者 "8" 或者 "9"

给你的表达式一定是一个expr，并且number总是 0到1,000,000,000之间的数。

思路：

　　这实际上就是一个计算器，只有两种运算： ">>" 这个事实上就是C语言里面的右位移，不过要注意的是，当移动的位数太大时，结果可能会向下溢出，因此要特判一下。
另外"S<>" 可以看成一种单目运算。

　　因为是计算器嘛，理所应当的想到用栈来做。 两个栈，一个存放运算数，另一个存放运算符。当遇到"S"或者">>"运算的时候，直接入栈，当遇到数字的时候，先检查栈顶是否为>>运算，如果是，则从数字栈里面再弹出一个数来运算，否则入栈。 当遇到"S<>"中的">"时， 从数字栈中弹出一个数字进行运算。

　　如何区分 ">>"和 “>” 根据">>"后面一定会跟着数字，或者类似数字的东西。

　　另外，我先把所有的空格去掉了，觉得这样更好判断符号。

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <stack>

using namespace std;

typedef long long ll;

const ll MOD = (ll)1e9+7;
const int MAXN = (int)2e6+9;

stack<ll> num;
stack<char> op;

char str[MAXN];
char str2[MAXN];

inline ll S(ll x) {
return (x*x)%MOD;
}

void pushNum(ll x) {
while (!op.empty() && '^'==op.top()) { //其实这个用if就好了，因为>>运算时先计算左边，所以栈中不会出现连续的>>运算
op.pop();
ll tmp = num.top(); num.pop();
if (x>60) {
tmp = 0;
x = 0;
}
x = (tmp>>x);
}
num.push(x);
}

void slove() {
int i = 0;
ll tmp;
while (true) {
if ('\0'==str[i]) break;

while (' '==str[i]) i++;

if ('S'==str[i]) {
op.push('S');
i++;
continue;
}
if ('<'==str[i]) {
i++;
continue;
}
if ('0'<=str[i]&&str[i]<='9') {
tmp = 0;
while ('0'<=str[i]&&str[i]<='9') {
tmp *=10;
tmp += str[i]-'0';
i++;
}
pushNum(tmp);
continue;
}
if ('>'==str[i]) {
if ('>'==str[i+1] && ('S'==str[i+2] || ('0'<=str[i+2]&&str[i+2]<='9'))) {
op.push('^');
i++;
}else if ('S'==op.top()) {
op.pop();
tmp = num.top(); num.pop();
pushNum(S(tmp));
}
i++;
continue;
}
}
printf("%lld\n", num.top());
}

void Transform(){
int i = 0, j = 0;
while (true) {
if ('\0'==str2[i]) break;
if (' '==str2[i]) while (' '==str2[i]) i++;
else {
str[j++] = str2[i++];
}
}
str[j] = '\0';
}

int main()
{
#ifdef Phantom01
freopen("CSU1378.txt", "r", stdin);
#endif // Phantom01

while (gets(str2)) {
if ('#'==str2[0]&&'\0'==str2[1]) break;

while (!num.empty()) num.pop();
while (!op.empty()) op.pop();

Transform();
slove();
}

return 0;
}