Java使用FST实现地址逆向解析到区划信息

import java.util.HashMap;
import java.util.Map;

public class FSTNode {
    private Map<Character, FSTNode> transitions = new HashMap<>();
    private boolean isFinal;
    private String output;

    public FSTNode() {
        this.isFinal = false;
        this.output = null;
    }

    public void addTransition(char c, FSTNode nextNode) {
        transitions.put(c, nextNode);
    }

    public FSTNode getNextNode(char c) {
        return transitions.get(c);
    }

    public boolean isFinal() {
        return isFinal;
    }

    public void setFinal(boolean isFinal) {
        this.isFinal = isFinal;
    }

    public String getOutput() {
        return output;
    }

    public void setOutput(String output) {
        this.output = output;
    }
}

public class FST {
    private FSTNode root;

    public FST() {
        this.root = new FSTNode();
    }

    // 插入一个键值对
    public void insert(String key, String value) {
        FSTNode current = root;
        for (char c : key.toCharArray()) {
            FSTNode nextNode = current.getNextNode(c);
            if (nextNode == null) {
                nextNode = new FSTNode();
                current.addTransition(c, nextNode);
            }
            current = nextNode;
        }
        current.setFinal(true);
        current.setOutput(value);
    }

    // 查找一个键的值
    public String search(String key) {
        FSTNode current = root;
        for (char c : key.toCharArray()) {
            current = current.getNextNode(c);
            if (current == null) {
                return null; // 未找到
            }
        }
        return current.isFinal() ? current.getOutput() : null;
    }
}

public class AddressLookup {
    private FST fst;

    public AddressLookup() {
        this.fst = new FST();
        // 插入区划信息
        fst.insert("北京市", "110000");
        fst.insert("上海市", "310000");
        fst.insert("广州市", "440100");
        fst.insert("深圳市", "440300");
    }

    // 根据地址查询区划信息
    public String lookup(String address) {
        // 从地址中提取区划信息（这里简化逻辑）
        String[] regions = {"北京市", "上海市", "广州市", "深圳市"};
        for (String region : regions) {
            if (address.contains(region)) {
                return fst.search(region);
            }
        }
        return null; // 未找到
    }

    public static void main(String[] args) {
        AddressLookup lookup = new AddressLookup();

        String address1 = "广东省深圳市南山区科技园";
        String address2 = "上海市浦东新区张江高科技园区";
        String address3 = "北京市海淀区中关村";

        System.out.println("Address: " + address1 + " -> Region Code: " + lookup.lookup(address1));
        System.out.println("Address: " + address2 + " -> Region Code: " + lookup.lookup(address2));
        System.out.println("Address: " + address3 + " -> Region Code: " + lookup.lookup(address3));
    }
}

Address: 广东省深圳市南山区科技园 -> Region Code: 440300
Address: 上海市浦东新区张江高科技园区 -> Region Code: 310000
Address: 北京市海淀区中关村 -> Region Code: 110000