1. 程式人生 > 其它 >華為機試題: IPv6地址壓縮

華為機試題: IPv6地址壓縮

華為校招機試題。

題目

IPv6 的地址格式如下:
XXXX:XXXX:XXXX:XXXX:XXXX:XXXX:XXXX:XXXX的格式,它是128位的,用":"分成8段,每段包含4個X是一個16進位制數(0-9,A-F),全部大寫,如下就是一個IPv6地址:

IPv6地址一般較長,IPv6提供了一些規則來壓縮地址,規則如下:
-規則1:丟棄前導零。
-規則2:如果兩個或者多個塊包含連續零,則省略它們,並用::替換。連續的零塊只能被::替換一次,優先優化左側的。如果地址中仍有零塊,它們可以縮小到一個零。

測試用例

輸入 2001:0000:3238:DFE1:0063:0000:0000:FEFB
輸出 2001:0:3238:DFE1:63::FEFB

輸入 0000:0000:0000:0000:0000:0000:0000:0000
輸出 ::

程式碼

import java.util.Arrays;
import java.util.Scanner;
import java.util.StringJoiner;

public class Main {

    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        while (in.hasNext()) {
            String rawIpv6 = in.next();
            System.out.println(compressIpv6(rawIpv6));
        }
    }

    private static String compressIpv6(String rawIpv6) {
        if (rawIpv6.length() != 39) {
            return "error";
        }
        int[] segments = Arrays.stream(rawIpv6.split(":")).mapToInt(s -> Integer.valueOf(s, 16)).toArray();

        int startIndex = -1;
        int endIndex = -1;
        for (int i = 0; i < segments.length; i++) {
            if (segments[i] == 0) {
                if (startIndex < 0) {
                    // 第一次出現
                    startIndex = i;
                } else {
                    endIndex = i;
                }
            } else if (endIndex - startIndex < 1) {
                startIndex = endIndex = -1;
            }
        }
        StringJoiner joiner = new StringJoiner(":");
        for (int i = 0; i < segments.length; i++) {
            if (i >= startIndex && i < endIndex) {
                // 跳過的部分
            } else if (i == endIndex) {
                joiner.add("");
            } else {
                joiner.add(Integer.toHexString(segments[i]) + "");
            }
        }
        String result = joiner.toString().toUpperCase();
        return "".equals(result) ? "::" : result;
    }
}