結果

問題 No.3128 Isosceles Triangle
ユーザー ks2m
提出日時 2025-04-25 22:40:56
言語 Java
(openjdk 23)
結果
AC  
実行時間 772 ms / 2,500 ms
コード長 1,098 bytes
コンパイル時間 4,134 ms
コンパイル使用メモリ 81,076 KB
実行使用メモリ 103,552 KB
最終ジャッジ日時 2025-04-25 22:41:16
合計ジャッジ時間 17,472 ms
ジャッジサーバーID
(参考情報)
judge4 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff #

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;

public class Main {
	public static void main(String[] args) throws Exception {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		int n = Integer.parseInt(br.readLine());
		String[] sa = br.readLine().split(" ");
		int[] a = new int[n];
		Map<Integer, Integer> map = new HashMap<>();
		for (int i = 0; i < n; i++) {
			a[i] = Integer.parseInt(sa[i]);
			map.put(a[i], map.getOrDefault(a[i], 0) + 1);
		}
		br.close();

		Arrays.sort(a);
		long ans = 0;
		for (int k : map.keySet()) {
			int v = map.get(k);
			if (v >= 2) {
				int r = binarySearch(a, k * 2 - 1);
				int c1 = r + 1 - v;
				long c2 = (long) v * (v - 1) / 2;
				ans += c1 * c2;
			}
		}
		System.out.println(ans);
	}

	static int binarySearch(int[] array, int val) {
		int ng = array.length;
		int ok = -1;
		while (Math.abs(ok - ng) > 1) {
			int mid = (ok + ng) / 2;
			if (array[mid] <= val) {
				ok = mid;
			} else {
				ng = mid;
			}
		}
		return ok;
	}
}
0