1596:Bug Hunt

Bug Hunt

我的思路：对于每行code，若无等号，直接初始化数组即可，否则依次检查等号左边和右边的表达式是否正确。对于左边的表达式，只需检查到首个数组之前即可，因为可能要对其初始化；对于右边表达式则要全部检查。

对于复合表达式要如何检查呢？我的做法是先找出数字下标，确定其最内层数组，然后依次向外找出外层数组。

此外还有一些小细节需要特别注意，这次是一次AC，开心。具体代码如下：

#include<iostream>
#include<map>
#include<cstring>
using namespace std;
map<string,map<int,int> >arrset; //记录每个数组初始化后的元素
map<string,int>arrsize; //记录数组的size
int findDigit(string& s,int rp){
    int i = rp; //右边起始点在等号后面一位
    while(!isdigit(s[i])) i++;
    return i;
}
int stod(string& s,int i){
    int r = 0;
    while(s[i] && isdigit(s[i])){ r = r*10 + s[i] - '0'; i++; }
    return r;
}
int findArr(string& s,int ae){
    while(ae >= 0 && s[ae] != '[' && s[ae] != '=') ae--; //检查右边的话结束标志是'='
    return ae + 1;
}
int main(){
    string s,arr,arr2;
    while(getline(cin,s) && s[0] != '.'){
        arrset.clear();
        arrsize.clear();
        int bugline = 0,cnt = 0;
        while(s[0] != '.'){
            if(bugline){ getline(cin,s);continue; }
            cnt++;
            int dp = findDigit(s,0); //出现数字下标的位置
            int index = stod(s,dp);
            if(s.find('=') == string::npos){
                arrsize[s.substr(0,dp-1)] = index;
                getline(cin,s);
                continue;
            }
            int as,ae = dp - 2; //arr名字起点和终点下标
            as = findArr(s,ae);
            arr = s.substr(as,ae - as + 1);
            while(as >= 0){ //检查左边
                if(index >= arrsize[arr] || (as && !arrset[arr].count(index))){ //如果是首个数组的话不需要检查元素是否初始化
                    bugline = cnt;
                    cout<<cnt<<endl;
                    break;
                }
                if(!as) break;
                ae = as - 2;
                as = findArr(s,ae);
                index = arrset[arr][index];
                arr = s.substr(as,ae - as + 1);
            }
            if(bugline){ getline(cin,s); continue;}
            int rp = s.find('=') + 1; //等号右边开始的位置
            dp = findDigit(s,rp);
            int index2 = stod(s,dp);
            if(dp == rp){ arrset[arr][index] = index2; getline(cin,s); continue; }
            ae = dp - 2;
            as = findArr(s,ae);
            string arr2 = s.substr(as,ae - as + 1);
            while(as >= rp){ //检查右边
                if(index2 >= arrsize[arr2] || !arrset[arr2].count(index2)){
                    bugline = cnt;
                    cout<<cnt<<endl;
                    break;
                }
                if(as == rp) break;
                ae = as - 2;
                as = findArr(s,ae);
                index2 = arrset[arr2][index2];
                arr2 = s.substr(as,ae - as + 1);
            }
            if(bugline){ getline(cin,s); continue; }
            arrset[arr][index] = arrset[arr2][index2];
            getline(cin,s);
        }
        if(!bugline) cout<<"0\n";
    }
    return 0;
}

看了别人的代码发现其实题目中说数组名是单个字符而不是字符串，修改如下：

#include<bits/stdc++.h>
using namespace std;
map<char,map<int,int> >arrset; //记录每个数组初始化后的元素
map<char,int>arrsize; //记录数组的size
int findDigit(string& s,int rp){
    int i = rp; //右边起始点在等号后面一位
    while(!isdigit(s[i])) i++;
    return i;
}
int stod(string& s,int i){
    int r = 0;
    while(s[i] && isdigit(s[i])){ r = r*10 + s[i] - '0'; i++; }
    return r;
}
int main(){
    string s;
    char arr,arr2;
    while(getline(cin,s) && s[0] != '.'){
        arrset.clear();
        arrsize.clear();
        int bugline = 0,cnt = 0;
        while(s[0] != '.'){
            if(bugline){ getline(cin,s);continue; }
            cnt++;
            int dp = findDigit(s,0); //出现数字下标的位置
            int index = stod(s,dp);
            int ap = dp - 2; //arr名的位置
            arr = s[ap];
            if(s.find('=') == string::npos){
                arrsize[arr] = index;
                getline(cin,s);
                continue;
            }
            while(ap >= 0){ //检查左边
                if(index >= arrsize[arr] || (ap && !arrset[arr].count(index))){ //如果是首个数组的话不需要检查元素是否初始化
                    bugline = cnt;
                    cout<<cnt<<endl;
                    break;
                }
                if(!ap) break;
                ap -= 2;
                index = arrset[arr][index];
                arr = s[ap];
            }
            if(bugline){ getline(cin,s); continue;}
            int rp = s.find('=') + 1; //等号右边开始的位置
            dp = findDigit(s,rp);
            int index2 = stod(s,dp);
            if(dp == rp){ arrset[arr][index] = index2; getline(cin,s); continue; }
            ap = dp - 2;
            arr2 = s[ap];
            while(ap >= rp){ //检查右边
                if(index2 >= arrsize[arr2] || !arrset[arr2].count(index2)){
                    bugline = cnt;
                    cout<<cnt<<endl;
                    break;
                }
                if(ap == rp) break;
                ap -= 2;
                index2 = arrset[arr2][index2];
                arr2 = s[ap];
            }
            if(bugline){ getline(cin,s); continue; }
            arrset[arr][index] = arrset[arr2][index2];
            getline(cin,s);
        }
        if(!bugline) cout<<"0\n";
    }
    return 0;
}