2025. Line Fighting

C语言

#include<stdio.h>

int main()
{
    int t = 0, n = 0, k = 0;
    scanf("%d", &t);
    while (t--) {
        scanf("%d%d", &n, &k);
        int ret = n * (n - 1) / 2, g = n / k, x = n % k;
        if (n % k == 0) ret -= k * g * (g - 1) / 2;
        else ret -= x * (g + 1) * (g) / 2 + (k - x) * g * (g - 1) / 2;
        printf("%d\n", ret);
    }
    return 0;
}

C++

#include<iostream>
using namespace std;

int main()
{
    int t = 0, n = 0, k = 0;
    cin >> t;
    while (t--) {
        cin >> n >> k;
        int ret = n * (n - 1) / 2, g = n / k, x = n % k;
        if (n % k == 0) ret -= k * g * (g - 1) / 2;
        else ret -= x * (g + 1) * (g) / 2 + (k - x) * g * (g - 1) / 2;
        cin >> ret;
    }
    return 0;
}

贪心算法：

要求结果最大，减的次数越少越好，也就是每组人之间打架次数越少越好，于是尽量将组平分，若是无法平衡，也要尽量平分（相差1）

时间复杂度：o(1)

主要取决于输入的测试用例数量 t，对于每个测试用例，代码执行的是常数级别的运算，所以时间复杂度为 O(1)，由于仅涉及简单的数学计算，故时间复杂度较低。