HFUTOJ problem 1344实现总结

题目

求和
Time Limit: 1000 MS Memory Limit: 65536 KB
Total Submissions: 603 Accepted: 98

Description

对于正整数n,k，我们定义这样一个函数f，它满足如下规律
f(n,k=1)=-1+2-3+4-5+6…n
f(n,k=2)=-1-2+3+4-5-6…n
f(n,k=3)=-1-2-3+4+5+6…n
f(n,k=4)=-1-2-3-4+5+6+7+8…n
现在给出n和k，你的任务就是要计算f(n,k)的值。

Input

首先是一个整数T，表示有T组数据
接下来每组数据是n和k(1<=n,k<=100000000)

Output

打印出f(n,k)的值，每个输出单独占一行

Sample Input

3
1 1
2 1
3 1

Sample Output

-1
1
-2

Source

安徽省2015年“京胜杯”大学生程序设计竞赛

思路

这题实际上就是个给定总长和循环节长度的循环数组求和。数列总长为n，每个循环节长度为2k，一个循环节的值为k^2，只要算出循环节的数量，然后将最后一个可能为完成的循环节的值算出来即可

代码

#include <iostream>
#include <cmath>
using namespace std;
int main (void){
    int TestTimes = 0;
    cin >> TestTimes;
    while (TestTimes--){
        int n = 0, k = 0;
        cin >> n >> k;
        int LoopTimes = n / (k * 2);
        long long result = LoopTimes * pow(k, 2);
        for (int num = (LoopTimes * 2 * k) + 1; num <= n; num++){
            if (num - (LoopTimes*2*k) <= k)
                result -= num;
            else
                result += num;
        }
        cout << result << endl;
    }
    return 0;
}