CCF 2017-09-3 JSON查询

在看题之前想到的是用对象保存一个String，如果是对象保存到Map中，看了其他博主的方法感觉思路比我之前的好，也照着这个思路去实现，虽然思路清楚了，但是提交并没有AC，只有70分,也不晓得哪里出错了，测试了三层四层的嵌套也是没有问题的.

解题思路：
将输入的字符串拼接转义字符，空格替换，进入递归函数，判断是否值为对象的情况
1. 存在值为Object的情况
处理前缀的拼接(加”.”)，把整个对象用“Object”替换,然后正常分割两次放进map，然后递归调用函数，参数为Object替换的字符串，直到字符串中没有值为Object的情况则结束递归
2. 不存在值为Object的情况：
将字符串用split(“,”)分割,键值再分割一遍，然后put进map

这里判断值是否包含Object用的是正则贪婪匹配，避免对象中包含对象的情况

import java.util.HashMap;
import java.util.Map;
import java.util.Scanner;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

/**
 * @author Shixiaodong
 * @date 2018/3/16 20:47
 */

public class Main {

    //转义字符，空格替换
    public static String replaceWorld(String target) {
        target = target.replace(" ", "").replace("\\\\", "\\").replace("\\\"", "\"");
        return target.substring(1, target.length() - 1);
    }

    //去掉字符串首尾双引号字符
    public static String getWorld(String target) {
        int start = target.indexOf("\"");
        int end = target.lastIndexOf("\"");
        return target.substring(start + 1, end);
    }

    //查找对象的前缀
    public static String getPrefix(int start, String target) {
        int found = 0, begin = 0, end = 0;
        for(int i = start - 1; i >= 0; i--) {
            if(target.charAt(i) == '\"') {
                found++;
                if(found == 1) {
                    end = i;
                } else if(found == 2) {
                    begin = i;
                }
            }
        }
        return target.substring(begin + 1, end);
    }

    public static Map<String, Object> map = new HashMap<String, Object>();

    public static void get(String prefix, String target) {
        //前缀处理
        if(prefix != null && !prefix.equals("")) {
            prefix = prefix + ".";
        }
        target = replaceWorld(target);
        String rule = "(\\{.*\\})";
        Pattern pattern = Pattern.compile(rule);
        Matcher matcher = pattern.matcher(target);
        int i = 0;
        if(!target.contains("{")) {
            //值不包含对象
            String[] json = target.split(",");
            for(int j = 0; j < json.length; j++) {
                String[] temp = json[j].split(":");
                map.put(prefix + getWorld(temp[0]), getWorld(temp[1]));
            }
            return;
        }
        while (matcher.find()) {
            int start = matcher.start();    //找到起始位置，然后向前找到前缀
            int end = matcher.end();
            //System.out.println(getPrefix(start, target));
            i++;
            target = new StringBuilder(target).replace(start, end, "\"OBJECT\"").toString();    //将对象替换为OBJECT字符串

            String[] json = target.split(",");
            for(int j = 0; j < json.length; j++) {
                String[] temp = json[j].split(":");
                map.put(prefix + getWorld(temp[0]), getWorld(temp[1]));
            }
            //matcher.group(i)不是匹配的分组
            get(prefix + getPrefix(start, target), matcher.group());    //拼接好前缀，递归调用，根据题意80%不超过2层
        }
    }

    public static Scanner scanner = new Scanner(System.in);
    public static void main(String[] args) {
        String[] temp = scanner.nextLine().split(" ");
        int rowNum = Integer.parseInt(temp[0]);
        int queryNum = Integer.parseInt(temp[1]);

        //去除格式，编成一行字符串处理
        StringBuilder cache = new StringBuilder();
        for (int i = 0; i < rowNum; i++) {
            cache.append(scanner.nextLine());
        }
        get("", cache.toString());


        String key;
        for (int i = 0; i < queryNum; i++) {
            key = scanner.nextLine();
            if(!map.containsKey(key)) {
                System.out.println("NOTEXIST");
                continue;
            }
            if(map.get(key).equals("OBJECT")) {
                System.out.println("OBJECT");
            } else {
                System.out.println("STRING " + map.get(key));
            }
        }
    }
}

官方测试用例：
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”

多层测试用例：
21 5
{
“firstName”:”John”,
“lastName”:”Smith”,
“address”:{
“streetAddress”:”2ndStreet”,
“sxd”:{
“name”:”Shixiaodong”,
“age”:”22”,
“aa”: {
“bb”: “cc”,
“dd”: “dd”,
“ee”: {
“rr”: “rr”
},
},
},
“city”:”NewYork”,
“state”:”NY”
},
“esc\aped”: “\”hello\””
}
address.sxd.aa.ee
address.sxd.aa.ee.rr
address.sxd.aa.dd
address.sxd
address.sxd.aa
样例输出:
OBJECT
STRING rr
STRING dd
OBJECT
OBJECT