Atcoder Beginner Content 143 D - Triangles

Calendar Clock iconCalendar Clock icon

atcoder

目次

# 問題

https://atcoder.jp/contests/abc143/tasks/abc143_d

# 解説

すべての辺の組み合わせを単純に調べるとO(N3)O(N^3)となり間に合わない.
なので、最後の1辺の決定を高速化してO(N2LogN)O(N^2 LogN)に収める.

長さに制約のある問題なので、すべての辺を長さが短い順にソートしておく.
AとBを決定するとCの辺のとれる長さの範囲が決定する.
この範囲に収まる長さのCの数をカウントすれば良い.
辺はソート済みなので、二分探索を2回行い、Cを満たす最小値と最大値を探しその間の要素数をカウントする.

# 計算量

O(N2logN)O(N^2logN)

※AとBを列挙するためにO(N2)O(N^2)、Cを二分探索で探すためにO(logN)O(logN)

# 解答

// C++ 14
#include <iostream>
#include <string>
#include <vector>
#include <list>
#include <algorithm>
#include <queue>
#include <stack>
#include <set>
#include <map>
#include <unordered_map>
#include <math.h>

#define ll long long
#define Int int
#define loop(x, start, end) for(Int x = start; x < end; x++)
#define loopdown(x, start, end) for(int x = start; x > end; x--)
#define rep(n) for(int x = 0; x < n; x++)
#define span(a,x,y) a.begin()+x,a.begin()+y
#define span_all(a) a.begin(),a.end()
#define len(x) (x.size())
#define last(x) (*(x.end()-1))

using namespace std;
#define MAX_N 2001

Int N;
vector<Int> L(MAX_N, -1);

void input() {
  cin >> N;
  loop(i,0,N) cin >> L[i];
}

void solve() {
  sort(span(L,0,N));  // sort edges by length asc
  
  Int count = 0;
  //O(N^2)
  loop(a,0,N-2) {
    loop(b,a+1,N) {
      Int min_c = L[a] - L[b] + 1;
      Int max_c = L[a] + L[b] - 1;
      // Binary search twice. O(2LogN)
      count += upper_bound(span(L,b+1,N), max_c) - lower_bound(span(L,b+1,N), min_c);
    }
  }
  cout << count << endl;
}

int main() {
  input();
  solve();
  return 0;
}

リモートフリーランス。ウェブサービス、スマホアプリエンジニア。
東アジアを拠点に世界を移動しながら活動してます!

お仕事のご依頼・お問い合わせはこちら

コメント