【题解】洛谷P2830 写程序

题目链接：P2830 写程序

题解

  按照题意模拟，分 $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;
}

总结

  较之于前面做的两题，难度是上来了，是题单中碰到的第一道需要花时间思考的题目。在理清了对变量嵌套的处理后，剩下的主要就是各种小细节了。感觉这题顶多算中等模拟吧。