华为OD 增强的strstr

题目描述

C 语言有一个库函数： char *strstr(const char *haystack, const char *needle) ，实现在字符串 haystack 中查找第一次出现字符串 needle 的位置，如果未找到则返回 null。

现要求实现一个strstr的增强函数，可以使用带可选段的字符串来模糊查询，与strstr一样返回首次查找到的字符串位置。

可选段使用“[]”标识，表示该位置是可选段中任意一个字符即可满足匹配条件。比如“a[bc]”表示可以匹配“ab”或“ac”。

注意目标字符串中可选段可能出现多次。

输入描述

与strstr函数一样，输入参数是两个字符串指针，分别是源字符串和目标字符串。

输出描述

与strstr函数不同，返回的是源字符串中，匹配子字符串相对于源字符串地址的偏移（从0开始算），如果没有匹配返回-1。

补充说明：源字符串中必定不包含‘[]’；目标字符串中‘[]’必定成对出现，且不会出现嵌套。

输入的字符串长度在[1,100]之间。

用例1

输入

abcd
b[cd]

输出

1

说明

相当于是在源字符串中查找bc或者bd，bc子字符串相对于abcd的偏移是1

import java.util.Scanner;
import java.util.*;

// 注意类名必须为 Main, 不要有任何 package xxx 信息
public class Main {

    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        String str = in.nextLine();
        String subStr = in.nextLine();
        boolean tag = false;
        int subLength = 0;
        StringBuilder builder = new StringBuilder();
        Map<Integer, Set<Character>> map = new HashMap<Integer, Set<Character>>();
        //统计子串长度和子串每个索引位置上的匹配集，后续以滑动窗口来匹配map。
        for (int i = 0; i < subStr.length(); i++) {
            if (subStr.charAt(i) == '[') {
                tag = true;
            } else if (subStr.charAt(i) == ']') {
                tag = false;
                subLength++;
            } else {
                Set<Character> set;
                if (map.get(subLength) == null) {
                    set = new HashSet<Character>();
                } else {
                    set = map.get(subLength);
                }
                set.add(subStr.charAt(i));
                map.put(subLength, set);
                if (!tag) {
                    subLength ++;
                }
            }
        }
        for (int i = 0; i + subLength< str.length(); i++) {
            String str1 = str.substring(i, i + subLength);
            boolean flag = checkStr(str1, map);
            if (flag) {
                System.out.println(i);
                return;
            }
        }
        System.out.println(-1);
    }

    public static boolean checkStr(String str, Map<Integer, Set<Character>> map) {
        int length = 0;
        for (int i = 0; i < str.length(); i++) {
            if (map.get(i).contains(str.charAt(i))) {
                length++;
            }
        }
        if (length == str.length()) {
            return true;
        } else {
            return false;
        }
    }
}