XTU-OJ 1477-前缀和

题目描述

你有两个数列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;
}