基础算法_模拟

题意：

给一个长度为n的数组A，我们现在要选出一个序列，要求就是这个序列的中位数最接近k，输出最长的那个满足要求的序列在数组A中的下标

输入：n和k，数组A

输出：最长的满足要求的序列的长度len，以及下标 p1, p2, …, plen (1 ≤ pi ≤ n, pi < pi + 1)

input
5 3
1 2 3 4 5
output
5
1 2 3 4 5

input
5 42
1 2 3 4 5
output
1
5

思路：

直接模拟就好了，但是最开始的时候好像我理解错了题意。

代码实现：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 2e5 + 5;
const ll mod = 1e9 + 7;
struct node{
	int val;
	int id;
}G[maxn];
bool cmp(node a,node b){
	return a.val < b.val;
}
int main(){
	int n,k;
	scanf("%d%d",&n,&k);
	int x;
	for(int i = 1;i <= n;i++){
		scanf("%d",&x);
		G[i].val = x;
		G[i].id = i;
	}
	sort(G + 1,G + 1 + n,cmp);
	int ans = 1e9;
	int len;
	int res;
	int id;
	for(int i = 1;i <= n;i++){
		if(abs(G[i].val - k) < ans){
			ans = abs(G[i].val - k);
			if(i <= n / 2) res = i * 2;
			else if(i == (n + 1) / 2) res = n;
			else res = (n - i) * 2 + 1;
			id = i;
		}
		else if(abs(G[i].val - k) == ans){
			if(i <= n / 2) len = i * 2;
			else if(i == (n + 1) / 2) len = n;//1 2 3 4,k = 3
			else len = (n - i) * 2 + 1;//选出来的序列是2 3 4，而不是
			if(len > res){//1 2 3 4
				res = len;
				id = i;
			}
		}
	}
	cout << res << endl;
	vector<int>vec;
	vec.clear();
	if(id <= (n + 1) / 2){
		for(int i = 1;i <= res;i++){
			vec.push_back(G[i].id);
		}
	}
	else{
		for(int i = n;i >= n - res + 1;i--){
			vec.push_back(G[i].id);
		}
	}
	sort(vec.begin(),vec.end());
	int tmp = vec.size();
	for(int i = 0;i < tmp;i++){
		printf("%d%c",vec[i],i == tmp - 1 ? '\n':' ');
	}
	return 0;
}