Uva1596 Bug Hunt 【map+stack】【习题5-9】

题目：Bug Hunt

题意：

输入并模拟执行一段程序，输出第一个bug所在的行。 每行程序有两种可能：
数组定义，格式为arr[size]。 定义之后所有元素均为未初始化状态。
赋值语句，格式为arr[index]=value。 例如a[0]=3或者a[a[0]]=a[1]。
赋值语句可能会出现两种bug：下标index越界；使用未初始化的变量（index和value都可能出现这种情况）。
思路：

利用map<pair<char,int>,int>模拟数组，map<char,int>用于存放定义数组的长度

定义语句不用考虑，直接模拟存放其长度即可

赋值语句要分别计算等号左边和右边，俩种情况：

（1）直接是数字：直接赋值即可（2）嵌套数组表达式：利用栈进行计算，由里向外逐层计算得出值！

在这期间，如果出现上述俩种bug直接return结束返回行数即可！

注意：只考虑上述俩种bug情况，其他的不用管！！！（这个题开始想的多了，搞了很多情况，搞了一天楞是没过！后来仔细看题后重敲了一遍A了。。。）

代码：

#include <iostream>
#include <map>
#include <set>
#include <stack>
#include <string>
using namespace std;
typedef  pair<char,int> position;//数组名+下标
map<position,int>arr;//存放数组的值
map<char,int>arrDefine;//存放数组的大小
bool judge(string str)//判断语句是定义还是赋值
{
    if(str.find("=") == string::npos)//没有找到，返回string::npos
        return true;
return false;}
int arrLen(string str)//计算数组的长度或下标值
{
    int len = 0;
    for(int i=0;str[i]!='\0';i++)
    {
        if(str[i] == ']') break;
        if(str[i] >='0' && str[i] <='9'){
        len *= 10;
        len += str[i]-'0';}
    }
return len;}
stack<char>sac;//用于计算嵌套数组
bool findValue(string str,int& index,int start)
{

    while(!sac.empty()) sac.pop();//清空栈
    if(str[start] >= '0' && str[start] <= '9')//直接是数字的
    {
        index = arrLen(str);
        return true;
    }
    else//是嵌套数组的
    {
        int index2 = 0;
        char name;
        for(int i=start;str[i]!='\0';)//将数组的名字入栈
        {

            if(str[i] == '[' || str[i] == ']') i++;
            else if(str[i] >= '0' && str[i] <= '9')
            {
                while(str[i] >= '0' && str[i] <= '9')
                {
                    index2 *= 10;
                    index2 += str[i]-'0';
                    i++;
                }
                break;
            }
            else
                sac.push(str[i++]);
        }
        while(!sac.empty())//开始由里向外计算值
        {
            name = sac.top();
            sac.pop();
            if(arrDefine[name] > index2)//没有越界的
            {
                if(arr.count(position(name,index2)) != 0)//已初始化的
                    index2 = arr[ position(name,index2) ];
                else return false;//未初始化的
            }
            else return false;//越界
        }
        index = index2;//将得到的赋值传回
    }
return true;}
bool solve(string str){
    if(judge(str))//定义
    {
        arrDefine[str[0]] = arrLen(str);//存放数组的大小
    }
    else//赋值
    {
        char arrName = str[0];
        int arrIndex;
        findValue(str,arrIndex,2);//计算下标
        if(arrIndex >= arrDefine[arrName]) return false;//越界
        int start = str.find("=") + 1,values;
        string tempStr = str.substr(start,str.length()-start);
        if(str[start] >= '0' && str[start] <= '9')//等于后面直接是数字
            values = arrLen(tempStr);
        else//等于后面是嵌套数组的
        {
            values = 0;
            if(!findValue(str,values,start)) return false;
        }
        arr[position(arrName,arrIndex)] = values;//数组赋值
    }
return true;}
int main()
{
    string str;
    int flag = 1,record;
    while(cin >> str && str[0]!='.')
    {
        flag = 1;
        int cot = 1;
        if(!solve(str))
        {
            flag = 0;
            record = cot;
        }
        while(cin >> str && str[0]!='.')
        {
            cot++;
            if(flag)
            {
                if(!solve(str))
                {
                    flag = 0;
                    record = cot;
                }
            }
        }
        if(flag)
            cout << "0\n";
        else
            cout << record <<endl;
        arr.clear();
        arrDefine.clear();
    }
    return 0;
}