华为机试-093-较难-HJ93.数组分组
一、描述
输入int型数组,询问该数组能否分成两组,使得两组中各元素加起来的和相等,并且,所有5的倍数必须在其中一个组中,所有3的倍数在另一个组中(不包括5的倍数),不是5的倍数也不是3的倍数能放在任意一组,可以将数组分为空数组,能满足以上条件,输出true;不满足时输出false。
1.1、输入描述
- 第一行是数据个数,第二行是输入的数据
1.2、输出描述
- 返回true或者false
二、示例
2.1、示例1
输入:
4
1 5 -5 1
输出:
true
说明:
第一组:5 -5 1
第二组:1
2.2、示例2
输入:
3
3 5 8
输出:
false
说明:
由于3和5不能放在同一组,所以不存在一种分法。
三、答案(java)
3.1、方法一
先把三和五的倍数都挑出来,算好两边的和sum3和sum5,所有数总和为sum,求出target = sum/2 - sum3,把剩余的数放在一起,转化成在一个list中找能不能凑出target。
package com.tzq.hwod;
import java.util.LinkedList;
import java.util.Scanner;
public class Main {
public static void main(String[] args) {
Scanner in = new Scanner(System.in);
while (in.hasNext()) {
LinkedList<Integer> list = new LinkedList<>();
int n = in.nextInt();
int sum5 = 0;
int sum3 = 0;
int sum = 0;
for (int i = 0; i < n; i++) {
int cur = in.nextInt();
if (cur % 5 == 0) {
sum5 += cur;
} else if (cur % 3 == 0) {
sum3 += cur;
} else {
list.add(cur);
}
sum += cur;
}
int target = sum / 2 - sum3;
if (sum % 2 != 0)
System.out.println("false");
else
System.out.println(helper(list, target));
}
}
private static boolean helper(LinkedList<Integer> list, int target) {
return helper(0, list, target);
}
private static boolean helper(int l, LinkedList<Integer> list, int target) {
if (l == list.size())
return target == 0;
return helper(l + 1, list, target - list.get(l)) || helper(l + 1, list, target);
}
}
四、答案(python 3)
4.1、方法一
#!/usr/bin/python
# -*- coding: UTF-8 -*-
def add(sm5,sm3,temp):
if len(temp)==0:
if sm5==sm3:
return True
else:
return False
else:
return add(sm5+temp[0],sm3,temp[1:]) or add(sm5,sm3+temp[0],temp[1:])
while True:
try:
n=int(input())
nums=list(map(int,input().split()))
num3=[]
num5=[]
temp=[]
for i in nums:
if i%5==0:
num5.append(i)
elif i%3==0:
num3.append(i)
else:
temp.append(i)
sm5,sm3=sum(num5),sum(num3)
a=add(sm5,sm3,temp)
if a:
print('true')
else:
print('false')
except:
break