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 = ;
 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) == )
     {
         for(int i =  ; i<n ; i++)
             scanf("%d" , col+i);
         for(int i =  ; i<n ; i++)
             scanf("%d" , v+i);
 
         int val = ;
         for(int i= ; 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 =  ; i<n ; i++){
             if(a[i] < ){
                 if(i == ){
                     n = ;
                     break;
                 }
                 if(-a[i] > a[i-]) n = i;
                 else n = i+;
                 break;
             }
         }
         if(m<=n){
             for(int i =  ; i<m ; i++)
                 val += a[i];
         }
         else{
             for(int i =  ; i<n ; i++){
                 val += a[i];
             }
             if((m-n)&) val -= a[n-];
         }
         printf("%d\n" , val);
     }
     return ;
 }