AirBnB春招筆試題

試題說明

筆試題只有一道，限時1小時。

模擬一個戰爭外交游戲，游戲中定義了三種操作:

A city1 Hold ： 軍隊A 占領了city1

A city1 Move city2 : 軍隊A從city1移動到city2

A city1 Support B ： 占領了city1的軍隊A增援軍隊B，得到增援的軍隊B的strength+1;

注意:

1. 每支軍隊的初始strength為1；

2. 如果有多個軍隊到達了同一個城市，則strength高的軍隊存活並占領該城市，strength低軍隊死亡；

3. 如果某支軍隊正在交戰（即其所在的城市除自己外還有其他軍隊），則該軍隊的任何增援行動無效；

輸入：

一系列字符串表示的的Action，如:

A Munich Hold
B Bohemia Move Munich
C Warsaw Support B

輸出：

按字母表順序排列的各個軍隊的狀態(死亡或者存活，如果存活輸出駐扎地點)，上面用例的輸出為:

A [dead]
B Munich
C Warsaw

下面是原題:

JAVA題解:

//游戲的規則是把所有的動作執行之后再判斷狀態，每個回合之中的所有動作沒有先后之分。
import java.util.*;

public class testAirBnB {
    static List<String> evaluateActions(List<String> actions) {
        Map<String, army> armies = new TreeMap();
        Map<String, city> cities = new HashMap<>();
        ArrayList<String[]> manu = new ArrayList<>();
        ArrayList<String[]> supportList = new ArrayList<>();

        for (String action : actions) {
            //將每個aciton拆解
            String[] s = action.split(" ");
            //注冊每個軍隊
            army a = new army(s[0]);
            armies.put(s[0], a);
            manu.add(s);
        }

        for (String[] str : manu) {
            //遍歷每個action, 執行每個action
            //獲取當前action的執行主體軍隊
            army a = armies.get(str[0]);
            //記錄該軍隊當前所在城市
            a.whereNow = str[1];
            //保存目前軍隊所在城市
            city tmpCity;

            //為a當前所在的城市注冊，如果記錄中不存在該城市，新建一個記錄，並將a軍隊加入其hold軍隊表
            if (!cities.containsKey(a.whereNow)) {//該城市之前未被注冊過
                tmpCity = new city(a.whereNow, a.strength, a.name);
                cities.put(a.whereNow, tmpCity);
            } else {
                //該城市之前注冊過，但因為之前的戰斗所有軍隊都死亡，現在可以重新hold；這里假設
                //所有的操作都是有效的，即不存在向已有軍隊駐扎的城市hold的action發生
                tmpCity = cities.get(a.whereNow);
                tmpCity.holdArmies.put(a.name, a.strength);
            }

            if (str[2].equals("Hold")) {
                //keep default;
            } else if (str[2].equals("Move")) {
                //從當前的城市撤退
                tmpCity.holdArmies.remove(a.name);
                //進軍到目的城市
                a.whereNow = str[3];
                //如果要去的城市已經被注冊，直接加入駐扎名單
                if (cities.containsKey(a.whereNow)) {
                    tmpCity = cities.get(a.whereNow);
                    tmpCity.holdArmies.put(a.name, a.strength);
                }
                //否則說明要去的城市未被駐扎過，直接注冊該城市並記錄駐扎信息
                else {
                    tmpCity = new city(a.whereNow, a.strength, a.name);
                    cities.put(a.whereNow, tmpCity);
                }


            } else if (str[2].equals("Support")) {
                //必須等到所有的軍隊移動完畢才能確定是否可以增援
                supportList.add(str);
//                //為友軍加buf
//                army friend = armies.get(str[3]);
//                friend.strength++;
//                //更新友軍所在城市注冊表中友軍的strength
//                city friendCity = cities.get(friend.whereNow);
//                friendCity.holdArmies.put(friend.name, friend.strength);
            }
        }

        //處理增援
        //如果A沒有underAttack, 則可以增援
        //否則不可以增援
        for(String[] sptAction: supportList){
            //如果A所在城市的駐扎軍隊大於1只，說明A正在underAttack, A的增援行動無效

            if(cities.get(armies.get(sptAction[0]).whereNow).holdArmies.size()>1)
                continue;
            //否則增援有效
            else{
                //為友軍加buf
                army friend = armies.get(sptAction[3]);
                friend.strength++;
                //更新友軍所在城市注冊表中友軍的strength
                city friendCity = cities.get(friend.whereNow);
                friendCity.holdArmies.put(friend.name, friend.strength);
            }
        }

        //跟據城市來判斷每只軍隊的生存狀況
        for (city tmpCity : cities.values()) {
            //當前城市的駐扎軍隊少於兩支，不會發生戰爭
            if (tmpCity.holdArmies.size() < 2)
                continue;
                //駐扎軍隊大於兩支

            else {
                List<Map.Entry<String, Integer>> list = new ArrayList<>(tmpCity.holdArmies.entrySet());
//                按treeMap的值從大到小排序
                Collections.sort(list, new Comparator<Map.Entry<String, Integer>>() {
                    @Override
                    public int compare(Map.Entry<String, Integer> o1, Map.Entry<String, Integer> o2) {
                        return -o1.getValue().compareTo(o2.getValue());
                    }
                });
                //如果該城市的駐扎軍隊表中戰力靠前的兩支軍隊的戰力相等，則駐扎在該城市的所有軍隊都回死亡(包括戰力更低的)
                if (list.get(0).getValue().equals(list.get(1).getValue())) {
                    //該城市所有軍隊都死亡
                    for (int i = 0; i < list.size(); i++) {
                        armies.get(list.get(i).getKey()).isAlive = false;
                    }
                } else {
                    //否則除了第一支軍隊（戰力最高的軍隊），其余軍隊死亡
                    for (int i = 1; i < list.size(); i++) {
                        armies.get(list.get(i).getKey()).isAlive = false;

                    }
                }
            }
        }

        // 循環檢查每個城市的狀態，如果isAlive為false, 輸出城市名+[dead]
        // 否則輸出城市名+whereNow
        List<String> result = new ArrayList<>();
        for (army unit : armies.values()) {
            //如果狀態標記為dead，直接輸出
            if (!unit.isAlive)
                result.add(unit.name + " " + "[dead]");
            else {
                result.add(unit.name + " " + unit.whereNow);
            }
        }

        for (String s : result)
            System.out.println(s);

        return result;
    }

    public static void main(String[] args) {

        List<String> input = new ArrayList<>();
        //測試數據1
//        input.add("A Munich Hold");input.add("B Warsaw Move Bohemia");
        //測試數據2
        input.add("A Munich Hold");input.add("B Bohemia Move Munich");input.add("C Warsaw Support B");
        //測試數據3
//        input.add("A Munich Hold");input.add("B Bohemia Move Munich");input.add("C Prussia Move Munich");input.add("D Warsaw Hold");
        //測試數據4
//        input.add("A Munich Support B");input.add("B Bohemia Move Prussia");input.add("C Prussia Hold");input.add("D Warsaw Move Munich");
        //測試數據5
//        input.add("A Munich Support B");input.add("B Oakland Move Munich");
        evaluateActions(input);
    }

    static class army {
        String name;
        String whereNow;
        int strength = 1;
//        boolean underAttack = false;
        boolean isAlive = true;

        army(String _name) {
            name = _name;
        }
    }

    static class city {
        String name;
        TreeMap<String, Integer> holdArmies = new TreeMap<>();

        city(String city_name, int strength, String army_name) {
            name = city_name;
            holdArmies.put(army_name, strength);
        }
    }

}