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

传送门: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]);
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值