Week4--作业3--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 非常想得到那只可爱的猫咪，你能帮帮他吗？

输入

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

输出

输出新数组 ans 的中位数

样例输入

样例输出

思路

综述

这道题是一道整数二分算法题；
去绝对值：将数组升序排列，在j>i的情况下，数值为正。
可以用于二分的情况：取值范围是单调的；
本题中，数P最小是0，最大是数组排序后，最大值减去最小值；
如果P小于中位数，则i,j对数小于正确值；
如果P等于中位数，则i,j对数等于正确值；
如果P大于中位数，则i,j对数大于正确值；

过程

注：P值只需要找到范围最大值最小值即可，不需要将每个P值列出来；
：二分答案P，遍历i二分j计算i,j对数
step1：输入数据
这里cin会超时

for (int i = 0; i < n; i++) {
			scanf("%d", &cat[i]);
			/*cin >> cat[i];*/
		}

step2:对数据排序并且计算P的取值范围

		sort(cat.begin(), cat.end());
		int l = 0;
		int r = cat[cat.size() - 1] - cat[0] ;

step3:计算，具体见下面注释
计算方法是，开辟一个数组，数组序号范围是P值的范围，如果序号小于中位数则该出赋值为1，否则赋值为0

//二分得到P值  mid
		while (l <= r) {
			int mid = (l + r) >> 1;
			int tot = 0;//用于记录ij的对数
			//遍历i值
			for (int i = 0; i < n - 1; i++) {
			//二分得到j
				int l1 = i+1;
				int r1 = n - 1;
				int ans = -1;
				//二分找j=mid1
				while (l1 <= r1) {
					int mid1 = (l1 + r1) >> 1;
					if (cat[mid1] <= cat[i] + mid) {
						ans = mid1 - i;
						l1 = mid1 + 1;
					}
					else {
						r1 = mid1 - 1;
					}
				}
				if (ans == -1)ans = 0;
				//else ans++;
				//记录i取一个值的情况下的j
				tot += ans;
			}
				if (tot >= ((n * (n - 1) / 2 + 1) / 2)) {
				v[mid] = 0; 
				r = mid - 1;
			}
			else {
				l = mid + 1;
			}
		}

step4：
遍历这个数组，找到第一个等于0的位置，输出序号

		for (int i = 0; i < v.size();i++) {
			if (v[i] != 1) {
				cout << i << endl;
				break;
			}
		}

总结

问题1：cin和scanf的问题
用cin这次会超时，而scanf时间大约500ms左右，这次用cin的时间消耗是scanf大约2倍；

代码

#include <iostream>
#include <queue>
#include <cstring>
#include <algorithm>
#include <cstdio>
#include <map> 
#include <vector>
using namespace std;
int n;

//int cat[1000000];
//int P[1000000];

int main() {
	while (cin >> n) {
		vector<int> cat(n);
		for (int i = 0; i < n; i++) {
			scanf("%d", &cat[i]);
			/*cin >> cat[i];*/
		}
		sort(cat.begin(), cat.end());
		int l = 0;
		int r = cat[cat.size() - 1] - cat[0] ;
		
		vector<int> v(r+1, 1);
		while (l <= r) {
			int mid = (l + r) >> 1;
			int tot = 0;
			for (int i = 0; i < n - 1; i++) {
				int l1 = i+1;
				int r1 = n - 1;
				int ans = -1;
				//二分找j=mid1
				while (l1 <= r1) {
					int mid1 = (l1 + r1) >> 1;
					if (cat[mid1] <= cat[i] + mid) {
						ans = mid1 - i;
						l1 = mid1 + 1;
					}
					else {
						r1 = mid1 - 1;
					}
				}
				if (ans == -1)ans = 0;
				//else ans++;

				tot += ans;
			}
				if (tot >= ((n * (n - 1) / 2 + 1) / 2)) {
				v[mid] = 0; 
				r = mid - 1;
			}
			else {
				l = mid + 1;
			}
		}
		for (int i = 0; i < v.size();i++) {
			if (v[i] != 1) {
				cout << i << endl;
				break;
			}
				
		}


	}
}