按字節截取字符串

題目描述

編寫一個截取字符串的函數，輸入爲一個字符串和字節數，輸出爲按字節截取的字符串。可是要保證漢字不被截半個，
如"我ABC"4，應該截爲"我AB"，輸入"我ABC漢DEF"6，應該輸出爲"我ABC"而不是"我ABC+漢的半個"。

輸入描述

輸入待截取的字符串及長度

輸出描述

截取後的字符串

輸入例子

我ABC漢DEF
6

輸出例子

我ABC

算法實現

import java.util.Scanner;

/**
 * All Rights Reserved !!!
 */
public class Main {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
//        Scanner scanner = new Scanner(Main.class.getClassLoader().getResourceAsStream("data.txt"));
        while (scanner.hasNext()) {
            String input = scanner.next();
            int n = scanner.nextInt();
            System.out.println(getStringByBytes(input, n));
        }

        scanner.close();
    }

    private static String getStringByBytes(String s, int n) {
        StringBuilder builder = new StringBuilder();

        for (int i = 0, sum = 0; i < s.length(); i++) {
            if (String.valueOf(s.charAt(i)).getBytes().length == 1) {
                sum += 1;
                builder.append(s.charAt(i));

                if (sum >= n) {
                    break;
                }
            } else {
                sum += 2;
                if (sum >= n) {
                    break;
                }
                builder.append(s.charAt(i));
            }
        }

        return builder.toString();
    }
}