容斥 + 组合数学 ---Codeforces Round #317 A. Lengthening Sticks

  Lengthening Sticks#

Problem's Link:  http://codeforces.com/contest/571/problem/A
#

---

 

Mean: 

给出a,b,c,l，要求a+x,b+y,c+z构成三角形，x+y+z<=l，成立的x,y,z有多少种。

analyse:

这题在推公式的时候细心一点就没问题了。

基本的思路是容斥：ans=所有的组合情况-不满足条件的情况。

1.求所有的组合情况

方法是找规律：

首先只考虑l全部都用掉的情况。

l=1：3

l=2：6

l=3：10

l=4：15

......

到这可能你会发现，其实l=i时，结果就是1+2+...+(i+1)，即：C(i+2,2)，也就是三角数。

然而i可以取0~l中的任何一个，那也很简单，一路累加上去就可。

2.不满足条件的情况：

三角形满足的条件是什么？任意两边之和大于第三边，那么不满足的必要条件就是第三边小于等于其它两边之和。

分别枚举a,b,c做第三边的情况，再考虑将剩下的l拆分三份分配给a,b,c依旧不满足的情况即可。

Time complexity: O(N)

 

Source code: 

/*
* this code is made by crazyacking
* Verdict: Accepted
* Submission Date: 2015-08-23-12.24
* Time: 0MS
* Memory: 137KB
*/
#include <queue>
#include <cstdio>
#include <set>
#include <string>
#include <stack>
#include <cmath>
#include <climits>
#include <map>
#include <cstdlib>
#include <iostream>
#include <vector>
#include <algorithm>
#include <cstring>
using namespace std;
typedef long long LL;

LL cal(LL a,LL b,LL c,LL l)
{
   LL ans=0;
   for(LL i=max(b+c-a,0LL);i<=l;++i)
   {
       LL x=min(l-i,a+i-b-c);
       ans+=(1+x)*(2+x)/2;
   }
   return ans;
}

int main()
{
     ios_base::sync_with_stdio(false);
     cin.tie(0);
     LL a,b,c,l;
   cin>>a>>b>>c>>l;
   LL ans=0;
   for(LL i=0;i<=l;++i)
       ans+=LL(1+i)*(2+i)/2;
   ans-=cal(a,b,c,l);
   ans-=cal(b,a,c,l);
   ans-=cal(c,a,b,l);
   cout<<ans;
     return 0;
}
/*

*/