COJ 1156 Switching bulbs

一道模拟题目

对于所有0 还是 1 我们都可以想象做均为 0 的状态

v[i]表示原来的值

但是对于原来为1的要加上其所在的值作为初始值

然后转化后 a[i] = -v[i]  ， 如果原来为0 ， 那就直接赋值

我们总是希望将尽可能大的值先加 ，所以将a由大到小排个序 ，  一直加到负数的时候需要考虑这个负数取不取，其实我们最多取一个负数，因为当加了

第一个负数若后面还为负数，那么我们可以将那个负数倒转又加回来即可

但第一个负数取不取也要看它的绝对值是不是比前一个数大，如果大，说明宁可将前一个数再变一次颜色后舍去也不需要加后一个负数

最后我们就确定实际上使用的数了，如果操作数大于这个数，说明超过范围只是不断的将最小的数不断改变颜色

#include <cstdio>
#include <algorithm>
using namespace std;

const int N = 100005;
int col[N] , v[N] , a[N];

bool cmp(int a , int b)
{
    return a>b;
}

int main()
{
   // freopen("a.in" , "r" , stdin);
    int n , m;
    while(scanf("%d%d" , &n , &m) == 2)
    {
        for(int i = 0 ; i<n ; i++)
            scanf("%d" , col+i);
        for(int i = 0 ; i<n ; i++)
            scanf("%d" , v+i);

        int val = 0;
        for(int i=0 ; i<n ; i++)
            if(col[i]){
                val += v[i];
                a[i] = -v[i];
            }
            else{
                a[i] = v[i];
            }
        sort(a , a+n , cmp);
        for(int i = 0 ; i<n ; i++){
            if(a[i] < 0){
                if(i == 0){
                    n = 1;
                    break;
                }
                if(-a[i] > a[i-1]) n = i;
                else n = i+1;
                break;
            }
        }
        if(m<=n){
            for(int i = 0 ; i<m ; i++)
                val += a[i];
        }
        else{
            for(int i = 0 ; i<n ; i++){
                val += a[i];
            }
            if((m-n)&1) val -= a[n-1];
        }
        printf("%d
" , val);
    }
    return 0;
}