1. 问题描述:
给定两个长度为 n 的 01 数组 a1,a2,…,an 和 b1,b2,…,bn。请你构造一个长度为 n 的正整数数组 p1,p2,…,pn。要求 ∑i=1nai×pi>∑i=1nbi×pi 成立。此外,maxi=1npi 需要尽可能小。输出最小可能值。
输入格式
第一行包含整数 n。第二行包含 n 个整数 a1,a2,…,an。第三行包含 n 个整数 b1,b2,…,bn。
输出格式
输出 maxi=1npi 的最小可能值。如果不存在满足条件的数组 p,则输出 −1。
数据范围
1 ≤ n ≤ 100,
0 ≤ ai,bi ≤ 1
输入样例1:
5
1 1 1 0 0
0 1 1 1 1
输出样例1:
3
输入样例2:
3
0 0 0
0 0 0
输出样例2:
-1
输入样例3:
4
1 1 1 1
1 1 1 1
输出样例3:
-1
输入样例4:
9
1 0 0 0 0 0 0 0 1
0 1 1 0 1 1 1 1 0
输出样例4:
4
来源:https://www.acwing.com/problem/content/description/3770/
2. 思路分析:
我们可以先整理一下对应的公式,可以发现当a[i] == b[i]的时候pi取什么值对结果都是没有什么影响的,而pi > 0所以令pi = 1这样可以使得pi最小,当a[i] < bi和a[i] > b[i]的时候统计一下对应的数目x和y,因为a[i] < b[i]所以要想使得结果大一点需要使得减去的数越小那么需要使pi = 1,最终∑pi > x,也即y * t > x(平均分配使得pi最小),最终可以得到下图中的式子,向上取整可以转化为向下取整:⌈x / k⌉ = ⌊(x + k - 1) / k⌋
3. 代码如下:
class Solution:
def process(self):
n = int(input())
a = list(map(int, input().split()))
b = list(map(int, input().split()))
x = y = 0
for i in range(n):
if a[i] < b[i]:
x += 1
elif a[i] > b[i]:
y += 1
if y == 0: return -1
# (x + y + 1 - 1) / y
return (x + y) // y
if __name__ == '__main__':
print(Solution().process())