Week4 作业

A - DDL 的恐惧

题目

ZJM 有 n 个作业，每个作业都有自己的 DDL，如果 ZJM 没有在 DDL 前做完这个作业，那么老师会扣掉这个作业的全部平时分。

所以 ZJM 想知道如何安排做作业的顺序，才能尽可能少扣一点分。

请你帮帮他吧！

Input

输入包含T个测试用例。输入的第一行是单个整数T，为测试用例的数量。

每个测试用例以一个正整数N开头(1<=N<=1000)，表示作业的数量。

然后两行。第一行包含N个整数，表示DDL，下一行包含N个整数，表示扣的分。

Output

对于每个测试用例，您应该输出最小的总降低分数，每个测试用例一行。

Sample Input

3
3
3 3 3
10 5 1
3
1 3 1
6 2 3
7
1 4 6 4 2 4 3
3 2 1 7 6 5 4

Sample Output

0
3
5

Hint

上方有三组样例。

对于第一组样例，有三个作业它们的DDL均为第三天，ZJM每天做一个正好在DDL前全部做完，所以没有扣分，输出0。

对于第二组样例，有三个作业，它们的DDL分别为第一天，第三天、第一天。ZJM在第一天做了第一个作业，第二天做了第二个作业，共扣了3分，输出3。

做法

观察数据范围：1<=N<=1000，数据范围不大，设计O（n^2）的算法即可。如果数据范围比较大，可以考虑用大根堆设计O（nlogn）算法。这里给出easy版做法。
考虑用结构体ddl数组a来储存截止时间和完成后对应的分数。每天只能完成一项作业，用数组mark标记当天是否完成ddl。输入分数的时候顺便把分数累加起来score，完成一项任务就减去对应的分数，最后score就是要扣的分。
接下来是贪心算法。把任务按分数降序排序，优先完成分数高的作业。完成作业时从作业ddl往前找空闲时间，这样可以保证尽量不影响其他任务；找到空闲时间就占用，更新mark，减去对应的分数，跳出循环，进行下一项任务的安排。

代码

#include<iostream>
#include<algorithm>
#include<string.h>
using namespace std;

struct ddl
{
	int d;//date
	int s;//score
}; 
bool cmp(ddl a,ddl b)
{
	return a.s>b.s;
}
int mark[1009];//标记当天是否完成ddl 

int main()
{
	int T;
	cin>>T;
	for(int j=0;j<T;j++)
	{
	memset(mark,0,1009);
	int n;
	cin>>n;
	ddl a[1009];
	for(int i=0;i<n;i++)
	cin>>a[i].d;
	int score=0;//总分
	for(int i=0;i<n;i++)
	{
		cin>>a[i].s;
		score=score+a[i].s;
	}
	sort(a,a+n,cmp);
	
	for(int i=0;i<n;i++)
	{
		for(int ii=a[i].d;ii>0;ii--)
		{
			if(mark[ii]==0)
			{
				score=score-a[i].s;
				mark[ii]=1;
				break;
			}
		}
	}
	cout<<score<<endl;
}
	return 0;	 
}

B-四个数列

题目

ZJM 有四个数列 A,B,C,D，每个数列都有 n 个数字。ZJM 从每个数列中各取出一个数，他想知道有多少种方案使得 4 个数的和为 0。

当一个数列中有多个相同的数字的时候，把它们当做不同的数对待。

请你帮帮他吧！

Input

第一行：n（代表数列中数字的个数） （1≤n≤4000）

接下来的 n 行中，第 i 行有四个数字，分别表示数列 A,B,C,D 中的第 i 个数字（数字不超过 2 的 28 次方）

Output

输出不同组合的个数。

Sample Input

6
-45 22 42 -16
-41 -27 56 30
-36 53 -37 77
-36 30 -75 -46
26 -38 -10 62
-32 -54 -6 45

Sample Output

5

Hint

样例解释: (-45, -27, 42, 30), (26, 30, -10, -46), (-32, 22, 56, -46),(-32, 30, -75, 77), (-32, -54, 56, 30).

做法

观察数据范围：1≤n≤4000，如果用四重循环枚举abcd中的每个数，时间复杂度是O（n^4），运行超时。
但是可以接受O（n^2）的复杂度，就可以换成两重循环枚举ab求和，然后枚举cd计算其相反数在ab中出现多少次，可以转换为计算一个数在有序数列中第一次和最后一次出现的位置。
继而可以利用分治算法，考虑用二分求解。两重循环求出a、b数组对应的和储存在ab数组中，排序。然后用两重循环计算c[i]+d[j]，循环中直接嵌套二分查找c[i]+d[j]的相反数在ab数组出现的第一个位置，可能不只有这一个，由于ab是有序数组，可以直接跟着一个while循环计算出现次数。

代码

#include<iostream>
#include<algorithm>
using namespace std;

int a[4009],b[4009],c[4009],d[4009];
int ab[16000009];

int main()
{
	int n;
	cin>>n;
	for(int i=0;i<n;i++)
	{
		cin>>a[i]>>b[i]>>c[i]>>d[i];
	}
	int k=0;//ab的索引 
	for(int i=0;i<n;i++)
	 for(int j=0;j<n;j++)
	 {
	 	ab[k]=a[i]+b[j];
	 	k++;
	 }
	sort(ab,ab+k); 
	 
	int num=0;//组合个数 
	
	for(int i=0;i<n;i++)
	 for(int j=0;j<n;j++)
	 {
	 	int l=0,r=k-1;
	 	while(l<=r)//二分查找c[i]+d[j]相反数在ab出现的第一个位置 
	 	{
	 		int mid=(l+r)>>1;
	 		if(-ab[mid]<=c[i]+d[j])
	 		{
	 			r=mid-1;
			 }
			else
			l=mid+1;
		 }
		while(-ab[l]==c[i]+d[j]&&l<k)
		{
			num++;
			l++;
		}
	 }
	 
	cout<<num<<endl;
	return 0;
}

C - TT 的神秘礼物

题目

TT 是一位重度爱猫人士，每日沉溺于 B 站上的猫咪频道。

有一天，TT 的好友 ZJM 决定交给 TT 一个难题，如果 TT 能够解决这个难题，ZJM 就会买一只可爱猫咪送给 TT。

任务内容是，给定一个 N 个数的数组 cat[i]，并用这个数组生成一个新数组 ans[i]。新数组定义为对于任意的 i, j 且 i != j，均有 ans[] = abs(cat[i] - cat[j])，1 <= i < j <= N。试求出这个新数组的中位数，中位数即为排序之后 (len+1)/2 位置对应的数字，’/’ 为下取整。

TT 非常想得到那只可爱的猫咪，你能帮帮他吗？

Input

多组输入，每次输入一个 N，表示有 N 个数，之后输入一个长度为 N 的序列 cat， cat[i] <= 1e9 , 3 <= n <= 1e5

Output

输出新数组 ans 的中位数

Sample Input

4
1 3 2 4
3
1 10 2

Sample Output

1
8

做法

3 <= n <= 1e5，数据范围比较大，采用两重二分求解，复杂度是O（n（logn）^2）
给定一个数P，判断它是不是中位数，可以计算P在数列中的名次，如果它的名次等于中位数，它就是中位数。可以二分求解。
计算P的名次，可以将X从小到大排列，计算Xj-Xi<=P的二元组对数即可。移项得Xj<=Xi+P，i<j。枚举下标i计算满足条件的下标j的个数，可以二分求解。
接下来是对于代码的解释。题目要求的是ans数组的中位数，这个数组最小值不会小于0，最大值是升序排列的cat[n-1]-cat[0]，ans的长度是（(n-1)n)/2，那么中位数的名次m就是(((n-1)n)/2+1)/2，然后可以通过二分找中位数，P也就是mid=（l+r）/2，比中位数名次小就往右半部分找，否则往左半部分找。
计算名次的时候用到了find函数，考虑到重复值，所以返回的是第一个cat[j]>p+cat[i]的位置，（返回l，当有重复值的时候，l是第一个大于x的位置，而m是最后一个小于等于x的位置），j的位置减去i的位置再减去1累加才是实际上p的名次。因为有重复值，所以计算的是最大名次。
计算出的名次如果等于中位数的名次，它可能也不是中位数，因为mid是由l，r计算出来的，它可能并不是数组ans中的数。

代码

#include<iostream>
#include<stdio.h>
#include<algorithm>
using namespace std;

int cat[1000000];

int find(int x,int right)
{//返回第一个>p+cat[i](x)的位置 
	int left=0,mid;
	while(left<=right)
	{
		mid=(left+right)/2;
		if(cat[mid]<=x)
		{
		    left=mid+1;
	    }
		else
		right=mid-1;
	}
	return left;
}

int  main()
{
	int n;
	while(cin>>n)
	{
		for(int i=0;i<n;i++)
		{
			scanf("%d",&cat[i]);
		}
		sort(cat,cat+n);
		int m=(((n-1)*n)/2+1)/2;//中位数在有序ans中的位置 
		int l=0,r=cat[n-1]-cat[0],mid; //ans最大元素r
		while(l<=r)
		{
			mid=(l+r)/2;
			int rank=0;
			for(int i=0;i<n;i++)
			{
				rank+=find(cat[i]+mid,n-1)-i-1;
			}
			if(rank<m)
			 l=mid+1;
			else
			 r=mid-1;
		}
		cout<<r<<endl;
    }	
    return 0;
}