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]);
}
}