51 Nod 2497 数三角形二分-CFANZ编程社区

小b有一个仅包含非负整数的数组a，她想知道有多少个三元组(i,j,k)，满足i<j<k且a[i],a[j],a[k]可能作为某个三角形的三条边的边长。

收起

输入

第一行输入一个正整数n，表示数组a中元素个数；
第二行n个非负整数，表示a中元素，以空格隔开；
其中0＜n≤1000，a中任意元素a[i]满足0≤a[i]≤1000。

输出

输出一个数，表示满足题意的三元组个数

输入样例

4
2 2 3 4

输出样例

3


很容易想到一个O(n^2 * logn)的解法，我们将原来数组排完序后，枚举两个较小的边，然后二分出最长边的范围即可；

#include
#include
#include
#include
#include
#include
using namespace std;
#define maxn 100005
#define rdint(x) scanf("%d",&x)
typedef long long ll;

int n;
int a[1002];
int main() {
  rdint(n);
  for (int i = 1; i <= n; i++)rdint(a[i]);
  int tot = 0;
  sort(a + 1, a + 1 + n);
//  int nn = unique(a + 1, a + 1 + n) - (a + 1);
  for (int i = 1; i < n; i++) {
    for (int j = i + 1; j < n; j++) {
      int l = j + 1;
      int r = n;
      int ans1 = -1;
      while (l <= r) {
        int mid = (l + r) / 2;
        if (a[mid] > (a[j] - a[i])) {
          r = mid - 1; ans1 = mid;
        }
        else {
          l = mid + 1;
        }
      }
      ans1 = upper_bound(a + 1 + j + 1, a + 1 + n, a[j] - a[i]) - (a + 1);
      l = j + 1; r = n;
      int ans2 = -1;
      ans2 = lower_bound(a + 1 + j + 1, a + 1 + n, a[i] + a[j]) - (a + 1);
//      cout << ans1 << ' ' << ans2 <<' '<< a[i] + a[j] << endl;
      if (ans1 == -1 || ans2 == -1)continue;
//      tot += (ans2 - ans1 + 1);

      
      else if (ans2 == ans1 && a[ans1] > a[i] + a[j]) {
        tot += 0;
      }
      else if (ans2 == ans1 && a[ans2] < a[i] + a[j]) {
//        cout << 2 << endl;
        tot += (ans2 - ans1 + 1);
      }
      else if (ans2 > ans1 && a[ans2]        tot += (ans2 - ans1 + 1);
      }
      else if (ans2 > ans1&& a[ans2] >= a[i] + a[j]) {
        tot += (ans2 - ans1);
      }
    }
  }
  printf("%d\n", tot);
//  system("pause");
}

EPFL - Fighting

51 Nod 2497 数三角形 二分

输入

输出

输入样例

输出样例

51 Nod 2497 数三角形二分