奶牛棒球(枚举与双指针）

农夫约翰的 N头奶牛排成一排，每头奶牛都位于数轴中的不同位置上。

它们正在练习投掷棒球。

农夫约翰观看时，观察到一组三头牛 (X,Y,Z)(X,Y,Z) 完成了两次成功的投掷。

牛 X 把球扔给她右边的牛 Y，然后牛 Y 把球扔给她右边的牛 Z。

约翰指出，第二次投掷的距离不少于第一次投掷的距离，也不超过第一次投掷的距离的两倍。

请计算共有多少组牛 (X,Y,Z)(X,Y,Z) 可能是约翰所看到的。

输入格式

第一行包含整数 N。

接下来 N 行，每行描述一头牛的位置。

输出格式

输出奶牛三元组 (X,Y,Z)(X,Y,Z) 的数量。

(X,Y,Z)(X,Y,Z) 需满足，Y 在 X 的右边，Z 在 Y 的右边，并且从 Y 到 Z 的距离在 [XY,2XY][XY,2XY] 之间，其中 XY表示从 X到 Y 的距离。

数据范围

3≤N≤1000
奶牛所在的位置坐标范围 [0,1e8]。

输入样例：

5
3
1
10
7
4

输出样例：

4

样例解释

四个可能的奶牛三元组为：1−3−7,1−4−7,4−7−10,1−4−101−3−7,1−4−7,4−7−10,1−4−10。

（来源：USACO 2013 December Contest Bronze）

这道题的数据范围是1000，我们需要依次枚举出x,y,z三个数，所以我们如果完全用枚举来做的话是会超时的，但这个时候，我们只要来进行一下简单的剪枝，就能保证数据通过。

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
using namespace std ;
int n ;
int a[1100] ;
int res ;
int main()
{
    cin >> n ;
    for(int i = 0 ; i < n ; i ++)
    {
        cin >> a[i] ;
    }
    sort(a , a + n) ; //sort函数是并排和快排，时间复杂度为o（logn）
    for(int i = 0 ; i <= n - 3 ; i ++) //枚举x ， 只需枚举到倒数第三个数即可
    {
        for(int j = i + 1 ; j <= n - 2 ; j ++)//枚举y，只需枚举到倒数第二个数即可
        {
            for(int k = j + 1 ; k <= n - 1 ; k ++)//枚举z
            {
                if(a[j] - a[i] <= a[k] - a[j] && a[k] - a[j] <= 2 * (a[j] - a[i]))
                {
                    res ++ ;//找到一个符合条件的数是+1
                }
                else if(a[k] - a[j] > 2 * (a[j] - a[i])) break ;//剪枝，不再枚举较大的z值
            }
        }
    }

    cout << res << endl;
    return 0 ;
}

双指针做法其实和剪枝思想差不多，是换了一种方法找出了满足x,y的z。

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
using namespace std ;
int n ;
int a[1100] ;
int res ;
int main()
{
    cin >> n ;
    for(int i = 0 ; i < n ; i ++)
    {
        cin >> a[i] ;
    }
    sort(a , a + n) ;
    for(int i = 0 ; i <= n - 3 ; i ++)
    {
        for(int j = i + 1 , l = j + 1 , r = j + 1 ; j <= n - 2 ; j ++)
        {
            while(l < n && a[l] - a[j] < a[j] - a[i]) l ++ ;//找到第一个满足关系的数的下标
            while(r < n && a[r] - a[j] <= 2 * (a[j] - a[i])) r ++ ;//最后一个满足关系的下一个数的下标

            res += r - l ;
        }
    }

    cout << res << endl;
    return 0 ;
}

详细看一下中间双指针的过程，当a[l] - a[j] < a[j] - a[i]时，l就继续前进，直到找到满足a[l] - a[j] = a[j] - a[i]的数，r同理，满足a[r] - a[j] <= 2 * (a[j] - a[i] )时就一直前进，直到找到第一个a[r] - a[j] > 2 * (a[j] - a[i] )的数。如图所示：

 这个时候不难得出，其间满足条件的数的个数即为 r - l 。