BZOJ 3721: PA2014 Final Bazarek

 

3721: PA2014 Final Bazarek

Time Limit: 20 Sec  Memory Limit: 128 MB
Submit: 645  Solved: 261
[Submit][Status][Discuss]

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

HINT

 

Source

鸣谢Jcvb

[Submit][Status][Discuss]

 

先对所有数字按照从大到小的顺序排序，然后维护前缀和，以及最小前缀（奇数/偶数），最大后缀（奇数/偶数）。

当询问K个的时候，如果sum[k]凑巧是个奇数，那么答案就是sum[k]，否则就考虑用一个未添加的奇数换一个已添加的偶数，或是用一个未添加的偶数换一个已添加的奇数，二者取其大。

 

#include <bits/stdc++.h>

template <class T>
inline void read(T &x) {
    char c = getchar(); x = 0;
    while (c < '0')
        c = getchar();
    while (c >= '0') {
        x = x*10 + c - '0';
        c = getchar();
    }
}

template <class T>
inline T Max(const T &a, const T &b) {
    return a > b ? a : b;
}

template <class T>
inline T Min(const T &a, const T &b) {
    return a < b ? a : b;
}

inline int cmp(const void *a, const void *b) {
    return - *(int *)a + *(int *)b;
}

typedef long long longint;

const int inf = 2e9 + 9;

const int siz = 1000005;

int n, m;
int num[siz];
int min[siz][2];
int max[siz][2];
longint sum[siz];

inline void prework(void) {
    qsort(num + 1, n, sizeof(int), cmp);
    
    memset(sum, 0, sizeof(sum));
    
    for (int i = 0; i < siz; ++i)
        min[i][0] = min[i][1] = inf,
        max[i][0] = max[i][1] = -inf;
    
    for (int i = 1; i <= n; ++i)
        sum[i] = sum[i - 1] + num[i];
    
    for (int i = 1; i <= n; ++i) {
        int a = num[i] & 1, b = a ^ 1;
        min[i][a] = num[i];
        min[i][b] = min[i - 1][b];
    }
    
    for (int i = n; i >= 1; --i) {
        int a = num[i] & 1, b = a ^ 1;
        max[i][a] = num[i];
        max[i][b] = max[i + 1][b];
    }
}

inline bool judge(int k) {
    return 
        (min[k][0] == inf || max[k + 1][1] == -inf)
    &&  (min[k][1] == inf || max[k + 1][0] == -inf);
}

inline void query(int k) {
    if (sum[k] & 1)
        printf("%lld\n", sum[k]);
    else if (judge(k))puts("-1");
    else printf("%lld\n", 
        sum[k] + Max(
            - min[k][0] + max[k + 1][1],
            - min[k][1] + max[k + 1][0])
        );
}

signed main(void) {
    read(n);
    
    for (int i = 1; i <= n; ++i)
        read(num[i]);
        
    prework();
        
    read(m);
    
    for (int i = 1, k; i <= m; ++i)
        read(k), query(k);
}

 

@Author: YouSiki