面试刷题第一天
给定一个arr,里面的数字都是0~9, 你可以随意使用arr中的数字,哪怕打乱顺序也行, 请拼出一个能被3整除的最大的数字,用str形式返回。 来自去哪儿网。
贪心思路的解题算法如下:
首先计算数组的累加和,记为sum,然后
(1)如果sum除3,余数为0,则结果就是将数组中的数从大到小拼起来就ok;
(2)如果sum除3,余数为1,需要删掉一个最小除3余数为1的数,如果没有,需要删掉两个最小除3余数为2的数,然后从大到小将数拼起来;
(3)如果sum除3,余数为2,需要删掉一个最小除3余数为2的数,如果没有,需要删掉两个最小除3余数为1的数,然后从大到小将数拼起来;
(4)如果以上均做不到,说明拼不成符合要求的字符串。
Java代码如下:
package com.company;
import java.util.ArrayList;
public class Main {
public static String max(int[] arr) {
if (arr == null || arr.length == 0) {
return "";
}
int mod = 0;
ArrayList<Integer> array = new ArrayList<>();
for (int num : arr) {
array.add(num);
mod += num;
mod %= 3;
}
if ((mod == 1 || mod == 2) && !remove(array, mod, 3 - mod)) {
return "";
}
if (array.isEmpty()) {
return "";
}
array.sort((a, b) -> b - a);
if (array.get(0) == 0) {
return "0";
}
StringBuilder builder = new StringBuilder();
for (int num : array) {
builder.append(num);
}
return builder.toString();
}
public static boolean remove(ArrayList<Integer> arr, int first, int second) {
if (arr.size() == 0) {
return false;
}
arr.sort((a, b) -> compare(a, b, second));
int size = arr.size();
if (arr.get(size - 1) % 3 == first) {
arr.remove(size - 1);
return true;
} else if (size > 1 && arr.get(size - 1) % 3 == second && arr.get(size - 2) % 3 == second) {
arr.remove(size - 1);
arr.remove(size - 2);
return true;
} else {
return false;
}
}
public static int compare(int a, int b, int s) {
int ma = a % 3;
int mb = b % 3;
if (ma == mb) {
return b - a;
} else {
if (ma == 0 || mb == 0) {
return ma == 0 ? -1 : 1;
} else {
return ma == s ? -1 : 1;
}
}
}
//生成随机数组
public static int[] randomArray(int len) {
int[] arr = new int[len];
for (int i = 0; i < len; i++) {
arr[i] = (int) (Math.random() * 10);
}
return arr;
}
//测试
public static void main(String[] args) {
int len = (int) (Math.random() * 10);
int[] arr = randomArray(len);
String ans = max(arr);
System.out.println(ans);
}
}