CF1355C Count Triangles

传送门

 

 

 

思路：我们需要满足 x + y > z  ,  x + z > y ,  y + z > x .因为 A <= X <=B <= Y <= C <= Z <= D，所以 X + Z > Y和 Y + Z > X明显一定满足，所以我们只需要确定X + Y > Z的个数了.X∈[A,B],Y∈[B,C]，则我们发现若X = A,则 X + Y = [A + B, A + C], X = A + 1, X + Y = [A + B + 1, A + C + 1].可以看出这就是区间+1操作，这样我们只需要枚举X的取值范围得到一个[L,R]，让该区间值+1，最后我们就可以得到X+Y=[A+B,B+C]中任意一个数值的方案数，然后得到一个前缀和数组P[i]，表示X+Y=[1,i]的总方案数.这样我们只需要枚举Z∈[C,D]，对于每个Z的答案就是P[B+C] - P[Z]，当然如果当前的Z>=B+C需要考虑。而上面的区间+1操作可以用差分来解决，或者线段树也可以。复杂度就是O（N）。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <queue>
#include <vector>
#include <cstring>
#include <functional>
#include <map>
#define LL long long
#define lson(rt) (rt << 1)
#define rson(rt) (rt << 1 | 1)
using namespace std;

const int N = 1e6 + 10;
long long x[N], y[N];

void solve ()
{
    int a, b, c, d;
    scanf("%d%d%d%d", &a, &b, &c, &d);
    for(int i = a; i <= b; ++i) {
        x[i + b] += 1;
        x[i + c + 1] -= 1;
    }
    int n = b + c;
    for(int i = 1; i <= n; ++i) x[i] += x[i - 1];
    for(int i = 1; i <= n; ++i) y[i] += y[i - 1] + x[i];/*
    for(int i = 1; i <= s; ++i) cout << x[i] << " ";
    cout << endl;
    for(int i = 1; i <= s; ++i) cout << y[i] << " ";
    cout << endl;*/
    long long tot = 0;
    for(int i = c; i <= d; ++i) {
        if(i >= b + c) break;
        tot += (LL)y[b + c] - y[i];
    }
    //cout << "tot = " << tot << endl;
    printf("%lld\n", tot);
}

int main ()
{
    solve();

    return 0;
}