尺取法练习3

组合三角形

【题目描述】：

桌面上凌乱地摆放着N个木棍，长度分别为{a1,a2…,ai,…an}，N<=6000，1<=ai<=100,000,000，选择三个木棍为边组成三角形，问有多少种不同的组合。三个木棍的不同排列，只能算一种组合。假设ai<=aj<=ak。

【输入描述】：

第一行，一个整数N。

第二行N个木棍的长度。

【输出描述】：

只有一个数表示能够组成三角形的组数，最后的结果可能比较大。

【样例输入】

7
3 5 7 4 9 5 1

【样例输出】

15

题目描述中这题的数据范围是N<=6000，因此用一般N^3的暴力算法是肯定过不了的。这时我们又想到了尺取法。首先枚举i,j,k，先将i固定在第一个位置不动，j=i+1,k=j+1，这样j和k就是一个尺取法。更新答案后再把i往后移一位，如此反复，最终可以得到答案。

AC代码:

#include<iostream>
#include<cstdio>
#include<algorithm>
#define ll long long
using namespace std;
int a[6005],n,k,t,i,j;
ll ans;
int main(){
	scanf("%d",&n);
	for(i=0;i<n;i++){
		scanf("%d",&a[i]);
	}
	sort(a,a+n);
	for(int t=0;t<n-2;t++){
		i=t+1;j=t+2;
		while(i<n-1){
			while(a[j]<a[t]+a[i]&&j<n){
				j++;
			}
			ans+=j-i-1;
			i++;
		}
	}
	cout<<ans<<endl;
	return 0;
}