BZOJ系列3721《PA2014 Final Bazarek》题解

Description

有n件商品，选出其中的k个，要求它们的总价为奇数，求最大可能的总价。

Input

第一行一个整数n(1<=n<=1000000)，表示商品数量。
接下来一行有n个整数，表示每件商品的价格，范围在[1,10^9]。
接下来一行有一个整数m(1<=m<=1000000)，表示询问数量。
接下来m行，每行一个整数k[i](1<=k[i]<=n)。

Output

对于每个询问，输出一行表示保证奇数的情况下最大的总价。若无法满足要求，输出-1。

Sample Input

4
4 2 1 3
3
2
3
4

Sample Output

7
9
-1

算法分析：贪心

思路：首先从大到小排序求前i项和sum[i]，

分两种情况：

1.sum[i]为奇数，直接输出。

2.sum[i]为偶数，则必然要把前i个数中的某个数替换使满足条件，

可以把前i个数中的最小奇数换为其余数中的最大偶数，

也可以把前i个数中的最小偶数换为其余数中的最大奇数，

这两种替换方法都可以满足题目要求，比较两种方法哪个优即可。

需要预处理出第i个数之前的最小偶数奇数以及之后的最大偶数奇数。

时间复杂度：O(N)。

代码如下：

#include<iostream>
#include<cmath>
#include<cstring>
#include<cstdio>
#include<algorithm>
#define INF 2000000000
using namespace std;
typedef long long ll;
int N,M,K;
struct dong
{
	int s,b1,b2,a1,a2;//s表示原数，b1、b2表示该节点前的最小偶数和最小奇数，a1、a2表示该节点后的最大偶数和最大奇数。
	ll sum;
}d[1000010];
bool cmp(dong aa,dong bb)
{
	return aa.s>bb.s;
}
void init()
{
	scanf("%d",&N);
	for(int i=1;i<=N;i++) scanf("%d",&d[i].s);
	sort(d+1,d+N+1,cmp);
	for(int i=1;i<=N;i++) d[i].sum=d[i].s+d[i-1].sum;
	d[0].b1=d[0].b2=INF;
	for(int i=1;i<=N;i++)
	{
		d[i].b1=d[i-1].b1;
		d[i].b2=d[i-1].b2;
		if(d[i].s&1) d[i].b2=min(d[i].s,d[i].b2);
		else d[i].b1=min(d[i].s,d[i].b1);
	}
	for(int i=N;i;i--)
	{
		d[i].a1=d[i+1].a1;
		d[i].a2=d[i+1].a2;
		if(d[i].s&1) d[i].a2=max(d[i].s,d[i].a2);
		else d[i].a1=max(d[i].s,d[i].a1);
	}
	scanf("%d",&M);
}
void work()
{
	for(int i=1;i<=M;i++)
	{
		scanf("%d",&K);
		if(d[K].sum&1) printf("%lld\n",d[K].sum);
		else 
		{
			ll ans=-1;
			if(d[K].b2!=INF&&d[K+1].a1) ans=max(ans,d[K].sum-d[K].b2+d[K+1].a1);
			if(d[K].b1!=INF&&d[K+1].a2) ans=max(ans,d[K].sum-d[K].b1+d[K+1].a2);
			printf("%lld\n",ans);
		}
	}
}
int main()
{
	init();
	work();
	return 0;
}