按字节截取字符串

题目描述

编写一个截取字符串的函数，输入为一个字符串和字节数，输出为按字节截取的字符串。可是要保证汉字不被截半个，
如"我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();
    }
}