题目描述
你有两个数列A={a1,a2,⋯,an}和B={b1,b2,⋯,bm}。你可以将B中的元素,移动到数列A的任意位置,得到一个新的数列C={c1,c2,⋯,cn+m}。
我们定义前缀和函数f(k)=∑ki=1ci,求最大k,使得f(k)<0。
比如A={−4,3,1}, B={4,2,1},我们可以将B中的元素移动A数列中,得到新的数列C={1,2,−4,4,3,1},此时k=3时,f(3)=−1,可以发现无论哪种移动方案,使得f(k)<0时,最大的k=3。所以结果为3。
输入格式
第一行是一个整数T(1≤T≤300),表示样例的个数。
每个样例的第一行是两个整数n,m, (1≤n,m≤1000);第二行是n个元素ai;第三行是m个元素bj;所有元素的绝对值都不超过105。
输出格式
依次每行输出一个样例的结果,为一个整数。如果不存在这样的k,使得f(k)<0,输出0。
样例输入
3 3 2 1 2 3 1 2 3 2 1 2 3 -1 -2 3 3 -4 3 1 4 2 1样例输出
0 3 3样例解释
第一个样例,无论我们怎么处理,前缀和都不小于0;
第二个样例,我们可以合并得到数列{−1,−2,1,2,3},此时f(3)=−2<0,所以结果为3。
解题思路: 题目说明B放入A中,要求前 k 个数 之和 < 0。 前缀和小于0,其实就说明不必在意具体插入哪个位置。B中小于0的数是一定可以放入的, 所以先预处理,把B中负数都插入A中,然后分别求 A 、B 的前缀和,然后判断遍历判断 A数组中前i个数,能插入多少个 B 数组中的数
AC代码:
#include <stdio.h>
#include <stdlib.h>
int A[1010], B[1010]={-1010};
int cmp(const void* p1, const void* p2){
return *(int *)p1 - *(int *)p2;
}
int check(int left,int right,int num)
{
if (left > right) left = right;
while (left < right)
{
int mid = (left+right+1) >> 1;
if (B[mid] <= num) left = mid;
else right = mid-1;
}
return left;
}
int main()
{
int T,n,m,len;
int len_B,len_C;
scanf("%d",&T);
while (T --)
{
len_C = 0;
scanf("%d %d",&n,&m);
for (int i = 1; i <= n; i ++) scanf("%d",&A[i]);
for (int i = 1; i <= m; i ++) scanf("%d",&B[i]);
qsort(B,m+1,4,cmp); // 对B数组排序
for (len_B = 1; B[len_B] < 0; len_B ++) A[1] += B[len_B]; // 把B数组负数部分都加入A数组中,同时找到不为负的下标:len_B
for (int i = 2; i <= n; i ++) A[i] += A[i-1]; // 对A求累加和
for (int i = len_B+1; i <= m; i ++) B[i] += B[i-1]; // 对B非负部分求累加和
for (int i = 1; i <= n; i ++)
{
if (A[i] >= 0) continue; // 如果A数组中 已经>0,则已经放不下B数组中的数了,跳过
len = check(len_B, m, -A[i]); // 找到B中还能放入个数的位置
if (B[len] >= -A[i]) len --;
if (i+len > len_C ) len_C = i+len;
}
printf("%d\n",len_C);
}
return 0;
}