CodeForces 571 A.Lengthening Sticks（组合数学）

Description

有三根长度分别为$a,b,c$的木棍，现在可以给每根木棍加长度，但加的总长不能超过$l$，问使得这三根木棍可以组成三角形的方案数

Input

四个整数$a,b,c,l(1\le a,b,c\le 10^5,0\le l\le 10^5)$

Output

输出方案数

Sample Input

1 1 1 2

Sample Output

4

Solution

设给三根木根加的长度分别为$aa,bb,cc$，则有$0\le aa+bb+cc\le l$，总方案数$C_{l+3}^3$(即为$l$个物品放到四个盒子的方案数)，现在考虑不合法情况，假设$a+aa\ge b+bb+c+cc$，枚举$aa$，则有$0\le bb+cc\le a-b-c+aa$，且注意到$0\le bb+cc\le l-aa$，令$t=min(a-b-c-aa,l-aa)$，那么问题转化为将$t$个物品放到三个盒子($bb$,$cc$和多余的)的方案数，由插板法知方案数为$C_{t+2}^2$，其他两种情况同理，时间复杂度$O(l)$

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=100001;
int main()
{
    int a,b,c,l;
    while(~scanf("%d%d%d%d",&a,&b,&c,&l))
    {
        ll ans=(ll)(l+1)*(l+2)*(l+3)/6;
        for(int i=0;i<=l;i++)
        {
            int t=min(a-b-c+i,l-i);
            if(t>=0)ans-=(ll)(t+1)*(t+2)/2;
            t=min(b-a-c+i,l-i);
            if(t>=0)ans-=(ll)(t+1)*(t+2)/2;
            t=min(c-a-b+i,l-i);
            if(t>=0)ans-=(ll)(t+1)*(t+2)/2;
        }
        printf("%I64d\n",ans);
    }
    return 0;
}