JSON查询

[CCF]JSON查询

Question

　　JSON (JavaScript Object Notation) 是一种轻量级的数据交换格式，可以用来描述半结构化的数据。JSON 格式中的基本单元是值 (value)，出于简化的目的本题只涉及 2 种类型的值：

• 字符串 (string)：字符串是由双引号 ” 括起来的一组字符（可以为空）。如果字符串的内容中出现双引号 “，在双引号前面加反斜杠，也就是用 \” 表示；如果出现反斜杠 \，则用两个反斜杠 \ 表示。反斜杠后面不能出现 ” 和 \ 以外的字符。例如：”“、”hello”、”\”\”。

• 对象 (object)：对象是一组键值对的无序集合（可以为空）。键值对表示对象的属性，键是属性名，值是属性的内容。对象以左花括号 { 开始，右花括号 } 结束，键值对之间以逗号 , 分隔。一个键值对的键和值之间以冒号 : 分隔。键必须是字符串，同一个对象所有键值对的键必须两两都不相同；值可以是字符串，也可以是另一个对象。例如：{}、{“foo”: “bar”}、{“Mon”: “weekday”, “Tue”: “weekday”, “Sun”: “weekend”}。
  除了字符串内部的位置，其他位置都可以插入一个或多个空格使得 JSON 的呈现更加美观，也可以在一些地方换行，不会影响所表示的数据内容。例如，上面举例的最后一个 JSON 数据也可以写成如下形式。

  “`json
  　　{

  　　“Mon”: “weekday”,

  　　“Tue”: “weekday”,

  　　“Sun”: “weekend”

  　　}

  “`

  给出一个 JSON 格式描述的数据，以及若干查询，编程返回这些查询的结果。

Input

第一行是两个正整数 n 和 m，分别表示 JSON 数据的行数和查询的个数。
　　接下来 n 行，描述一个 JSON 数据，保证输入是一个合法的 JSON 对象。
　　接下来 m 行，每行描述一个查询。给出要查询的属性名，要求返回对应属性的内容。需要支持多层查询，各层的属性名之间用小数点 . 连接。保证查询的格式都是合法的。

Output

对于输入的每一个查询，按顺序输出查询结果，每个结果占一行。
　　如果查询结果是一个字符串，则输出 STRING ，其中 是字符串的值，中间用一个空格分隔。
　　如果查询结果是一个对象，则输出 OBJECT，不需要输出对象的内容。
　　如果查询结果不存在，则输出 NOTEXIST。

样例输入

json
10 5
{
"firstName": "John",
"lastName": "Smith",
"address": {
"streetAddress": "2ndStreet",
"city": "NewYork",
"state": "NY"
},
"esc\\aped": "\"hello\""
}
firstName
address
address.city
address.postal
esc\aped


样例输出

STRING John
OBJECT
STRING NewYork
NOTEXIST
STRING "hello"

Solution

// http://118.190.20.162/view.page?gpid=T61
#include<cstdio>
#include<iostream>
#include<sstream>
#include<cstdlib>
#include<cstring>
#include<string>
#include<climits>
#include<cmath>
#include<algorithm>
#include<queue>
#include<vector>
#include<stack>
#include<set>
#include<cctype>
#include<map>
using namespace std;
#define print(A) cout << #A << ": "<< A << endl;
#define inf 0x3fffffff
typedef long long int ll;

string input, line;
map<string, vector<string> > mp;
int line_num, query_num;
bool isFirstBracket = true;
int myindex = 0;
// 每次遇到{就进行递归，遇到}结束递归
void dfs(string lastKey) {
  bool iskey = true;
  string key, value;
  // 先越过第一个{
  myindex++;
  for (; myindex < input.size(); myindex++) {
    // 找引号
    while (myindex < input.size() && input[myindex] != '\"' && input[myindex] != '{' && input[myindex] != '}') {
      // print(input[myindex]);
      myindex++;
    }
    if (myindex >= input.size()) return;
    // print(input[myindex]);
    if (input[myindex] == '\"') {
      myindex++;
      while (myindex < input.size() && input[myindex] != '\"') {
        if (input[myindex] == '\\')  {
          if (iskey) key.push_back(input[myindex+1]);
          else value.push_back(input[myindex+1]);
          myindex += 2;
          continue;
        }
        if (iskey) key.push_back(input[myindex]);
        else value.push_back(input[myindex]);
        myindex++;
      }
      if (iskey == false) {
        if (lastKey != "") mp[lastKey+"."+key].push_back(value);
        else mp[key].push_back(value);
        // print(key);
        // print(value);
        key.clear();
        value.clear();
      }
      iskey = !iskey;
    } else if (input[myindex] == '{') {
      if (lastKey != "") {
        mp[lastKey+"."+key].push_back("OBJECT");
        dfs(lastKey+"."+key);
      } else {
        mp[key].push_back("OBJECT");
        dfs(key);
        key.clear();
        value.clear();
      }

      iskey = true;
    } else if (input[myindex] == '}') {
      return;
    }
  }
}

int main(int argc, char const *argv[]) {
  cin >> line_num >> query_num;
  getchar();
  for (int i = 0; i < line_num; i++) {
    getline(cin, line);
    input += line;
  }

  bool iskey = true;
  string key, value;

  for (myindex = 0; myindex < input.size(); myindex++) {
    while (myindex < input.size() && input[myindex] != '{') myindex++;
    if (input[myindex] == '{') dfs("");
  }

  for (int i = 0; i < query_num; i++) {
    string query;
    cin >> query;
    //找到了
    if (mp[query].size() == 1 && mp[query][0] != "OBJECT") {
      cout << "STRING ";
      cout << mp[query][0] << endl;
    } else if (mp[query].size() == 0) {
      cout << "NOTEXIST" << endl;
    } else {
      cout << "OBJECT" << endl;
    }
  }
  return 0;
}

思路：一开始采取暴力破解的方法，强行进行解析，但是想到嵌套可能是无数层的，必须使用一个递归函数来解决。后来写出了一个递归函数dfs，注意一下细节问题就好了。