题目链接:P2830 写程序
题解
按照题意模拟,分
3
3
3 种情形处理不同的程序语句。稍微麻烦一点的是对嵌套变量的处理(例如 a[a[b[0]]]
这种),对此,可以采用类似这题的处理方法,递归地解析。访问数组过程中,需要特判数组下标越界的情况,具体细节可参考代码。
实现的代码见下,代码中使用了嵌套哈希表存储不同的数组信息(包括数组长度及其元素的值),同时在 <哈希表>[数组名][-1] 的位置存储数组的长度信息。
#include <iostream>
#include <string>
#include <unordered_map>
using namespace std;
string s1, s2;
unordered_map<string, unordered_map<int, int>> arrmp;
string getArrName(string & s, int & pos) {
string ans = "";
while (s[pos] != '[')
ans += s[pos++];
return ans;
}
int parseExp(string & s, int & pos) {
if (isdigit(s[pos])) {
int ans = 0;
while (isdigit(s[pos])) {
ans = ans * 10 + s[pos] - '0';
++pos;
}
return ans;
}
string arr = getArrName(s, pos);
++pos;
int val = parseExp(s, pos);
++pos;
if (val < 0 || val >= arrmp[arr][-1]) return -1;
return arrmp[arr][val];
}
int main() {
ios::sync_with_stdio(false);
cin.tie(nullptr);
int fg = 0;
while (cin >> s1 >> s2) {
if (fg) continue;
int pos1 = 0, pos2 = 0;
if (s1 == "int") {
string arr = getArrName(s2, pos2);
++pos2;
int val = parseExp(s2, pos2);
// 数组越界处理
if (val < 0) {
fg = 1;
cout << "-1\n";
continue;
}
// 存储数组长度信息
arrmp[arr][-1] = val;
} else if (s1 == "cout") {
int val = parseExp(s2, pos2);
// 数组越界处理
if (val < 0) {
fg = 1;
cout << "-1\n";
continue;
}
cout << val << '\n';
} else {
string arr1 = getArrName(s1, pos1);
++pos1;
int val1 = parseExp(s1, pos1), val2 = parseExp(s2, pos2);
// 数组越界处理
if (val1 < 0 || val2 < 0 || val1 >= arrmp[arr1][-1]) {
fg = 1;
cout << "-1\n";
continue;
}
arrmp[arr1][val1] = val2;
}
}
return 0;
}
总结
较之于前面做的两题,难度是上来了,是题单中碰到的第一道需要花时间思考的题目。在理清了对变量嵌套的处理后,剩下的主要就是各种小细节了。感觉这题顶多算中等模拟吧。