COJ 1156 Switching bulbs

一道模拟题目

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

v[i]表示原来的值

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

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

 

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

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

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

 

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

 1 #include <cstdio>

 2 #include <algorithm>

 3 using namespace std;

 4 

 5 const int N = 100005;

 6 int col[N] , v[N] , a[N];

 7 

 8 bool cmp(int a , int b)

 9 {

10     return a>b;

11 }

12 

13 int main()

14 {

15    // freopen("a.in" , "r" , stdin);

16     int n , m;

17     while(scanf("%d%d" , &n , &m) == 2)

18     {

19         for(int i = 0 ; i<n ; i++)

20             scanf("%d" , col+i);

21         for(int i = 0 ; i<n ; i++)

22             scanf("%d" , v+i);

23 

24         int val = 0;

25         for(int i=0 ; i<n ; i++)

26             if(col[i]){

27                 val += v[i];

28                 a[i] = -v[i];

29             }

30             else{

31                 a[i] = v[i];

32             }

33         sort(a , a+n , cmp);

34         for(int i = 0 ; i<n ; i++){

35             if(a[i] < 0){

36                 if(i == 0){

37                     n = 1;

38                     break;

39                 }

40                 if(-a[i] > a[i-1]) n = i;

41                 else n = i+1;

42                 break;

43             }

44         }

45         if(m<=n){

46             for(int i = 0 ; i<m ; i++)

47                 val += a[i];

48         }

49         else{

50             for(int i = 0 ; i<n ; i++){

51                 val += a[i];

52             }

53             if((m-n)&1) val -= a[n-1];

54         }

55         printf("%d\n" , val);

56     }

57     return 0;

58 }