CCF 201709-3 JSON查詢

題意：

問題描述

　　JSON (JavaScript Object Notation) 是一種輕量級的數據交換格式，可以用來描述半結構化的數據。JSON 格式中的基本單元是值 (value)，出於簡化的目的本題只涉及 2 種類型的值：
　　* 字符串 (string)：字符串是由雙引號 " 括起來的一組字符（可以為空）。如果字符串的內容中出現雙引號 "，在雙引號前面加反斜杠，也就是用 \" 表示；如果出現反斜杠 \，則用兩個反斜杠 \\ 表示。反斜杠后面不能出現 " 和 \ 以外的字符。例如：""、"hello"、"\"\\"。
　　* 對象 (object)：對象是一組鍵值對的無序集合（可以為空）。鍵值對表示對象的屬性，鍵是屬性名，值是屬性的內容。對象以左花括號 { 開始，右花括號 } 結束，鍵值對之間以逗號 , 分隔。一個鍵值對的鍵和值之間以冒號 : 分隔。鍵必須是字符串，同一個對象所有鍵值對的鍵必須兩兩都不相同；值可以是字符串，也可以是另一個對象。例如：{}、{"foo": "bar"}、{"Mon": "weekday", "Tue": "weekday", "Sun": "weekend"}。
　　除了字符串內部的位置，其他位置都可以插入一個或多個空格使得 JSON 的呈現更加美觀，也可以在一些地方換行，不會影響所表示的數據內容。例如，上面舉例的最后一個 JSON 數據也可以寫成如下形式。
　　{
　　"Mon": "weekday",
　　"Tue": "weekday",
　　"Sun": "weekend"
　　}
　　給出一個 JSON 格式描述的數據，以及若干查詢，編程返回這些查詢的結果。

輸入格式

　　第一行是兩個正整數 n 和 m，分別表示 JSON 數據的行數和查詢的個數。
　　接下來 n 行，描述一個 JSON 數據，保證輸入是一個合法的 JSON 對象。
　　接下來 m 行，每行描述一個查詢。給出要查詢的屬性名，要求返回對應屬性的內容。需要支持多層查詢，各層的屬性名之間用小數點 . 連接。保證查詢的格式都是合法的。

輸出格式

　　對於輸入的每一個查詢，按順序輸出查詢結果，每個結果占一行。
　　如果查詢結果是一個字符串，則輸出 STRING <string>，其中 <string> 是字符串的值，中間用一個空格分隔。
　　如果查詢結果是一個對象，則輸出 OBJECT，不需要輸出對象的內容。
　　如果查詢結果不存在，則輸出 NOTEXIST。

評測用例規模與約定

　　n ≤ 100，每行不超過 80 個字符。
　　m ≤ 100，每個查詢的長度不超過 80 個字符。
　　字符串中的字符均為 ASCII 碼 33-126 的可打印字符，不會出現空格。所有字符串都不是空串。
　　所有作為鍵的字符串不會包含小數點 .。查詢時鍵的大小寫敏感。
　　50%的評測用例輸入的對象只有 1 層結構，80%的評測用例輸入的對象結構層數不超過 2 層。舉例來說，{"a": "b"} 是一層結構的對象，{"a": {"b": "c"}} 是二層結構的對象，以此類推。

分析：

1、首先將輸入處理成無空格字符串。

2、因為每個對象中的鍵值互不相同。因此對於每個對象，記錄其內部鍵值的標號mp，內容ans，如果該鍵值對應的是對象，則記錄這個對象的標號ma。

3、然后根據對象的標號依次檢索答案即可。

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cctype>
#include<cmath>
#include<iostream>
#include<sstream>
#include<iterator>
#include<algorithm>
#include<string>
#include<vector>
#include<set>
#include<map>
#include<stack>
#include<deque>
#include<queue>
#include<list>
#define lowbit(x) (x & (-x))
const double eps = 1e-8;
inline int dcmp(double a, double b){
    if(fabs(a - b) < eps) return 0;
    return a > b ? 1 : -1;
}
typedef long long LL;
typedef unsigned long long ULL;
const int INT_INF = 0x3f3f3f3f;
const int INT_M_INF = 0x7f7f7f7f;
const LL LL_INF = 0x3f3f3f3f3f3f3f3f;
const LL LL_M_INF = 0x7f7f7f7f7f7f7f7f;
const int dr[] = {0, 0, -1, 1, -1, -1, 1, 1};
const int dc[] = {-1, 1, 0, 0, -1, 1, -1, 1};
const int MOD = 1e9 + 7;
const double pi = acos(-1.0);
const int MAXN = 8000 + 10;
const int MAXT = 10000 + 10;
using namespace std;
map<string, int> mp[MAXN];
map<int, int> ma[MAXN];
map<int, string> ans[MAXN];
int len;
string s;
int k;
vector<string> tmp;
bool vis[MAXN];
void solve(int fa, int &pos, int num);
void deal(int &i, int num, int cnt);
void deal(int &i, int num, int cnt){
    int len = s.size();
    int j, et, st;
    string x;
    for(j = i - 1; j >= 0; --j){
        if(s[j] == '\"'){
            et = j;
            break;
        }
    }
    for(j = et - 1; j >= 0; --j){
        if(s[j] == ',' || s[j] == '{'){
            break;
        }
        x += s[j];
    }
    x.resize(x.size() - 1);
    reverse(x.begin(), x.end());
    mp[num][x] = cnt;
    int id = mp[num][x];
    j = i + 1;
    if(s[j] == '\"'){
        st = j;
        x = "";
        for(j = st + 1; j < len; ++j){
            if(s[j] == ',' || s[j] == '}') break;
            x += s[j];
        }
        x.resize(x.size() - 1);
        ans[num][id] = "STRING " + x;
    }
    else if(s[j] == '{'){
        ans[num][id] = "OBJECT";
        ma[num][id] = ++k;
        solve(id, j, k);
    }
    i = j;
}
void solve(int fa, int &pos, int num){
    int cnt = 0;
    for(int i = pos + 1; i < len; ++i){
        if(s[i] == ':'){
            ++cnt;
            deal(i, num, cnt);
            if(s[i] == '}'){
                pos = ++i;
                break;
            }
        }
    }
}
bool Find(int cur, int num, int l, string& res, int id){
    if(num == l){
        res = ans[cur][id];
        return true;
    }
    else{
        cur = ma[cur][id];
        if(!mp[cur].count(tmp[num])) return false;
        return Find(cur, num + 1, l, res, mp[cur][tmp[num]]);
    }
}
int main(){
    int n, m;
    scanf("%d%d", &n, &m);
    getchar();
    while(n--){
        string str;
        getline(cin, str);
        int l = str.size();
        for(int i = 0; i < l; ++i){
            if(str[i] == ' ') continue;
            if(str[i] == '\\'){
                s += str[i + 1];
                ++i;
                continue;
            }
            s += str[i];
        }
    }
    len = s.size();
    for(int i = 0; i < len; ++i){
        if(s[i] == '{'){
            ++k;
            vis[k] = true;
            solve(-1, i, k);
        }
    }
    string x;
    while(m--){
        cin >> x;
        int l = x.size();
        tmp.clear();
        int id = 0;
        for(int i = 0; i < l; ++i){
            if(x[i] == '.'){
                tmp.push_back(x.substr(id, i - id));
                id = i + 1;
            }
        }
        tmp.push_back(x.substr(id, l - id));
        l = tmp.size();
        string res;
        bool ok = false;
        for(int i = 1; i <= k; ++i){
            if(!vis[i]) continue;
            if(mp[i].count(tmp[0])){
                int id = mp[i][tmp[0]];
                if(Find(i, 1, l, res, id)){
                    ok = true;
                    printf("%s\n", res.c_str());
                    break;
                }
            }
        }
        if(!ok){
            printf("NOTEXIST\n");
        }
    }
    return 0;
}