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;
}