結果

問題 No.742 にゃんにゃんにゃん　猫の挨拶
ユーザー 37zigen37zigen
提出日時 2018-10-05 22:50:14
言語 Java21
(openjdk 21)
結果
AC  
実行時間 391 ms / 2,500 ms
コード長 2,230 bytes
コンパイル時間 2,034 ms
コンパイル使用メモリ 79,300 KB
実行使用メモリ 48,696 KB
最終ジャッジ日時 2024-04-20 18:00:25
合計ジャッジ時間 5,269 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 119 ms
41,400 KB
testcase_01 AC 118 ms
41,620 KB
testcase_02 AC 109 ms
40,448 KB
testcase_03 AC 104 ms
40,412 KB
testcase_04 AC 130 ms
41,420 KB
testcase_05 AC 144 ms
41,776 KB
testcase_06 AC 159 ms
41,932 KB
testcase_07 AC 174 ms
42,848 KB
testcase_08 AC 192 ms
43,432 KB
testcase_09 AC 114 ms
41,428 KB
testcase_10 AC 106 ms
40,240 KB
testcase_11 AC 391 ms
48,696 KB
testcase_12 AC 343 ms
47,580 KB
testcase_13 AC 119 ms
41,572 KB
testcase_14 AC 121 ms
41,572 KB
testcase_15 AC 106 ms
40,528 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.io.FileNotFoundException;
import java.io.PrintWriter;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Scanner;

class Main {

	class BIT {
		int n;
		long[] v;

		public BIT(int n_) {
			n = n_;
			v = new long[n + 1];
		}

		void add(int k, int val) {
			while (k <= n) {
				v[k] += val;
				k += k & -k;
			}
		}

		long sum(int k) {
			long ret = 0;
			while (k >= 1) {
				ret += v[k];
				k -= k & -k;
			}
			return ret;
		}

		long sum(int l, int r) {
			return sum(r - 1) - sum(l - 1);
		}
	}

	void run() {
		Scanner sc = new Scanner(System.in);
		int N = sc.nextInt();
		int[] M = new int[N];
		SegmentTree seg1 = new SegmentTree(N);
		SegmentTree seg2 = new SegmentTree(N);
		SegmentTree seg3 = new SegmentTree(N);
		long ans = 0;
		for (int i = 0; i < N; ++i) {
			M[i] = sc.nextInt();
			--M[i];
			if (i < M[i]) {
				// 同じ方向
				ans += seg1.query(M[i], N);
				seg1.add(M[i]);
			} else if (i > M[i]) {
				// 同じ方向
				ans += seg2.query(M[i], i + 1);
				// 内側方向
				ans += seg1.query(M[i], N);
				seg2.add(M[i]);
			} else
				seg3.add(M[i]);
		}
		// 動かない猫との挨拶をカウント
		for (int i = 0; i < N; ++i) {
			if (i != M[i]) {
				ans += seg3.query(Math.min(i, M[i]), Math.max(i, M[i]) + 1);
			} else {
				ans += seg3.query(i, i + 1) - 1;
			}
		}
		PrintWriter pw = new PrintWriter(System.out);
		pw.println(ans);
		pw.close();
	}

	class SegmentTree {
		int n;
		int[] sum;

		public SegmentTree(int n_) {
			n = 1;
			while (n < n_) {
				n *= 2;
			}
			sum = new int[2 * n - 1];
		}

		void add(int k) {
			k += n - 1;
			++sum[k];
			while (k > 0) {
				k = (k - 1) / 2;
				sum[k] = sum[2 * k + 1] + sum[2 * k + 2];
			}
		}

		int query(int a, int b) {
			return query(0, n, a, b, 0);
		}

		int query(int l, int r, int a, int b, int k) {
			if (r <= a || b <= l)
				return 0;
			else if (a <= l && r <= b)
				return sum[k];
			else {
				return query(l, (l + r) / 2, a, b, 2 * k + 1) + query((l + r) / 2, r, a, b, 2 * k + 2);
			}
		}
	}

	void tr(Object... objects) {
		System.out.println(Arrays.deepToString(objects));
	}

	public static void main(String[] args) throws FileNotFoundException {
		new Main().run();
	}
}
0