传送门: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]);
}
}