codeforces 1301C Ayoub's function (排列组合)

 

Example

Input
5
3 1
3 2
3 3
4 0
5 2
Output
4
5
6
0
12

题意:给出01字符串的长度,告诉1的数量,求包含1区间的个数。
思路:先求出全部区间的数量,然后贪心的想用1把0全部平均分割开,然后减去只有0组合在一起的数量。
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
using namespace std;
#define PI 3.14159
#define ll long long
int t;
ll n, m;
int main()
{
    ll ans, cnt, tmp;
    scanf("%d", &t);
    while (t--)
    {
        scanf("%lld%lld", &m, &n);
        ans = m * (m + 1) / 2;
        ans -= m - n;
        if (n == 0)
        {
            printf("0\n");
            continue;
        }
        else if (n >= m / 2) {
            printf("%lld\n", ans);
            continue;
        }
        m -= n;
        cnt = m / (n + 1);
        tmp = m % (n + 1);
        ans -= cnt * (cnt - 1) / 2 * (n + 1) + tmp * cnt;
        cout << ans << endl;
    }
}

 

posted @ 2020-03-04 21:00  Tangent_1231  阅读(162)  评论(0编辑  收藏  举报