CodeForces 721D. Maxim and Array(贪心,思维题)

来源:互联网 发布:手机相机算法 编辑:程序博客网 时间:2024/05/18 13:30

传送门:http://codeforces.com/problemset/problem/721/D

题目大意:

输入三个整数n k x,接下来是一个长度为n 的数组
你可以至多进行k次操作,每次把数组里任意一个元素+x或者减x,
输出变化后能使得结果乘积最大的新数组。

题目分析:
抓住奇数个负数相乘为负,偶数个负数相乘为正这一原则设计贪心思路:
按绝对值对数组排序,然后每轮操作都讨论负数个数:
(1)若为偶数个,则把绝对值最小的那个往反方向变(正数尽量变负,负数尽量变正,0变负的,这样可以使得负数个数尽可能沿着变为奇数的趋势);
(2)若为奇数个,则把绝对值最小的那个往正负无穷大的方向变。
因为对于a,b两个数,假设a,b都为正数,且a<b,显然有a(b+x)<b(a+x),所以要变a,一开始变了b,WA了一发。

这个排序的维护可以用优先队列,我不小心写成了TreeSet,就这样吧。
而且c++的set也是红黑树实现维持有序,我以为是像java那样用Hash实现的,画蛇添足了。这次代码写的不是一般的烂,就不调整了。。。。

//Java真tm长啊!!!!!!!!!!import java.util.Comparator;import java.util.Scanner;import java.util.TreeSet;public class Main {    private static class MyClass {        public long data;        public int index;        public MyClass(long data, int index) {            this.data = data;            this.index = index;        }    }    public static void main(String[] args) {        Scanner sc = new Scanner(System.in);        int n=sc.nextInt();        int k=sc.nextInt();        int x=sc.nextInt();        long[] a=new long[n+5];        for(int i=0;i<n;i++)            a[i]=sc.nextLong();        TreeSet<MyClass> ts = new TreeSet<>(new Comparator<MyClass>() {            @Override            public int compare(MyClass o1, MyClass o2) {                long l=Math.abs(o1.data)-Math.abs(o2.data);                if(l>0)                    return 1;                else if(l<0)                    return -1;                else                    return o1.index-o2.index;            }        });        int minus=0;        for (int i = 0; i < n; i++) {            ts.add(new MyClass(a[i], i));            if(a[i]<0)                minus++;        }        while(k>0) {            if((minus&1) == 0) { //偶数个负数                MyClass c1 = ts.ceiling(new MyClass(0, 1));                if(c1.data==0) {                    int id=c1.index;                    a[id]-=x;                    minus++;                    ts.remove(c1);                    ts.add(new MyClass(a[id], id));                }                else { //没有0                    c1 = ts.first();                    int id = c1.index;                    if(a[id]<0) {                        a[id]+=x;                        if(a[id]>=0)                            minus--;                    }                    else {                        a[id]-=x;                        if(a[id]<0)                            minus++;                    }                    ts.remove(c1);                    ts.add(new MyClass(a[id], id));                }            }            else {                MyClass c1 = ts.ceiling(new MyClass(0, 1));                if(c1.data==0) {                    int id=c1.index;                    a[id]+=x;                    ts.remove(c1);                    ts.add(new MyClass(a[id], id));                }                else {                    c1 = ts.first();                    int id = c1.index;                    if(a[id]<0)                        a[id]-=x;                    else                        a[id]+=x;                    ts.remove(c1);                    ts.add(new MyClass(a[id], id));                }            }            k--;        }        for(int i=0;i<n-1;i++) {            System.out.print(a[i]+" ");        }        System.out.println(a[n-1]);    }}
0 0