798D Mike and distribution

最新推荐文章于 2021-06-18 09:32:50 发布

AndrewThompson

最新推荐文章于 2021-06-18 09:32:50 发布

阅读量354

点赞数

分类专栏： Codeforces

本文链接：https://blog.youkuaiyun.com/zju2016/article/details/70738735

版权

Codeforces 专栏收录该内容

32 篇文章

订阅专栏

这是一个使用C++编写的程序，主要功能是从两个输入序列中选择元素并进行特定的排序操作。程序首先读取一个整数n，然后分别读取两个长度为n的整数序列A和B。接下来，程序按A序列的降序排列对序列的索引进行排序，并根据特定条件从序列B中选择元素。最终输出被选中元素的数量及其在原始序列中的位置。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include<iostream>

#include<vector>
#include<string>
#include<set>
#include<map>
#include<algorithm>
#include<queue>
using namespace std;

int main(){
	int n;
	while (cin >> n){
		vector<int> A(n);
		vector<int> B(n);
		vector<int> index(n);
		for (int i = 0; i < n; i++){
			index[i] = i;
			cin >> A[i];
		}
		for (int i = 0; i < n; i++) cin >> B[i];
		sort(index.begin(), index.end(), [&](int a, int b)->bool {return A[a] > A[b]; });
		vector<int> res;
		res.push_back(index[0]);
		for (int i = 1; i < n; i += 2){
			int temp = index[i];
			if (i + 1 < n&&B[temp] < B[index[i + 1]]) temp = index[i + 1];
			res.push_back(temp);
		}
		cout << res.size() << endl;
		for (int i = 0; i < res.size(); i++) cout << res[i] + 1 << " ";
		cout << endl;
	}
	return 0;
}