【算法系列学习】codeforces D. Mike and distribution 二维贪心

时间:2023-11-11 22:50:20

http://codeforces.com/contest/798/problem/D

http://blog.csdn.net/yasola/article/details/70477816

对于二维的贪心我们可以先让它变成其中一维有序,这样只需要重点考虑另一维,就会简单很多。

首先,对于题目要求的选择元素之和两倍大与所有元素之和,我们可以转化为选择元素之和大于剩下的。然后我们可以将下标按照a从大到小排序。然后选择第一个,之后每两个一组,选择b大的一个,如果n是偶数再选择最后一个。

至于这样写的正确性:首先对于数组b,每一组选择的都是大的,而且还有多选的,所以一定比剩下的大。对于数组a,从第一个开始看,当前选择的,一定比下一组剩下的a大。所以这样贪心就一定正确。

 #include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<algorithm>
#include<cmath> using namespace std;
const int maxn=1e5+;
struct node
{
int x;
int id;
}a[maxn],b[maxn];
int n;
bool cmp(const node& aa,const node& bb)
{
return aa.x>bb.x;
}
int main()
{
scanf("%d",&n);
for(int i=;i<n;i++)
{
scanf("%d",&a[i].x);
a[i].id=i+;
}
for(int i=;i<n;i++)
{
scanf("%d",&b[i].x);
b[i].id=i+;
}
printf("%d\n",n/+);
sort(a,a+n,cmp);
printf("%d",a[].id);
for(int i=;i<n;i+=)
{
if(b[a[i].id-].x>b[a[i+].id-].x)
{
printf(" %d",a[i].id);
}
else
{
printf(" %d",a[i+].id);
}
}
printf("\n");
return ;
}