【腾讯2018实习笔试题】求周期性改变元素符号的数组的和

题目描述：

输入格式为一行两个数，比如8 2，则表示为这样一个数组{-1,-2,3,4,-5,-6,7,8}，-1-2+3+4-5-6+7+8 = 8，输出格式为求这个数组的和。

比如输入7 1，表示一个数组{-1,2,-3,4,-5,6,-7}

注意，起始元素都是-1开始。

解题方案：

这是腾讯18年实习笔试的第一题，很简单，用辅助3个变量在时间复杂度bigO(n)内就能解决。

代码：

import java.util.Scanner;

public class FirstQuestion {

	public static void main(String[] args) {
		Scanner in = new Scanner(System.in);
		
		long num = in.nextInt();
		int arr = in.nextInt();
		
		int sum = 0;
		int a=0;
		int flag = -1;
		for(int i=1; i <= num; i++){
			sum += i*flag;
			a++;
			if(a == arr){
				flag = -flag;
				a=0;
			}
		}
		
		System.out.println(sum);
	}
}

测试结果：