Description
万圣节又到了!FJ打算带他的奶牛去参加一个化装晚会,但是FJ只做了一套能容下三头总长不超过 S 的牛的恐怖服装。FJ养了 N 头按 1..N 顺序编号的奶牛,编号为 i 的奶牛的长度为 Li。如果三头奶牛的总长度不超过 S,那么她们就能穿下这套服装。
FJ想知道,如果他想选择三头不同的奶牛来穿这套衣服,一共有多少种满足条件的方案。
Input
第 1 行是 2 个用空格隔开的整数 N 和 S; 接下来有 n 行,每行包含一个整数 L_i,表示奶牛的身长为。
Output
仅一行一个整数,表示FJ可选择的所有方案数。注意奶牛顺序不同的两种方案是被视为相同的。
Sample Input 1
5 10
3 2 5 1 7
Sample Output 1
5
Hint
50%的数据满足:1<=n<=500
100%的数据满足:1<=n<=2000,1<=S<=2,000,000,000,1<=L_i<=2,000,000,000
上来先看到的是数据爆int,所以开long long
注意到n的数据量2000,(O(n^2))是安全的,所以可以枚举前两头牛,第三头牛排序后二分搜索,复杂度(O(n^2logn))
相比原版化装晚会,牛数由2头变成了3头,不能保证当前两头牛的长度和一定不大于前面的两头牛之和,所以不能再用贪心前后指针移动来解决。
#include <cstdio>
#include <algorithm>
#define ll long long
using namespace std;
ll L[2005];
int main() {
ll n,s;
ll ans=0;
scanf("%lld%lld",&n,&s);
for(int i=1; i<=n; i++)
scanf("%lld",&L[i]);
sort(L+1,L+1+n);
for(ll i=1; i<=n; i++) {
for(ll j=i+1; j<=n; j++) {
ll q2=L[i]+L[j];
if(q2<=s) {
ll p=upper_bound(L+1+j,L+1+n,s-q2)-L;
ans+=p-j-1;
}
}
}
printf("%lld",ans);
}