ホーム AtCoder Typical 90 014 — 昔は一緒に歌を歌ったものだ
記事
キャンセル

AtCoder Typical 90 014 — 昔は一緒に歌を歌ったものだ

問題リンク

N 個の整数を含む 2 つの配列が与えられます。1 つ目の配列の各要素を 2 つ目の配列の要素 1 つと組み合わせ、絶対差の合計を最小化します。

両方の配列を昇順に並べ、同じ添字の要素同士を組にすると最小値になります。x <= y かつ u <= v である 2 つずつの値を考えます。同じ順序で組にする費用は |x - u| + |y - v|、交差して組にする費用は |x - v| + |y - u| です。数直線上では、順序どおりに結ぶ場合の合計距離は交差させる場合より大きくなりません。したがって、交差した 2 組は入れ替えても費用が増えず、これを繰り返すと、両配列の同じ順位同士を組にした形になります。

2 つの配列のソートに O(N log N)、ソート後の全要素の走査に O(N) かかるため、全体の時間計算量は O(N log N) です。配列に O(N) の領域を使います。2 つの int の差は Integer.MAX_VALUE を超えることがあるため、減算前に long に変換し、合計も long で保持します。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
import java.io.*;
import java.util.*;

public class Main {
  static class FastScanner {
    private final InputStream input;
    private final byte[] buffer = new byte[1 << 16];
    private int length = 0;
    private int pointer = 0;

    FastScanner(InputStream input) {
      this.input = input;
    }

    private int read() throws IOException {
      if (pointer == length) {
        length = input.read(buffer);
        pointer = 0;
        if (length == -1) return -1;
      }
      return buffer[pointer++];
    }

    int nextInt() throws IOException {
      int c;
      do {
        c = read();
      } while (c <= ' ' && c != -1);

      int sign = 1;
      if (c == '-') {
        sign = -1;
        c = read();
      }
      int value = 0;
      while (c > ' ') {
        value = value * 10 + c - '0';
        c = read();
      }
      return sign * value;
    }
  }

  public static void main(String[] args) throws IOException {
    FastScanner input = new FastScanner(System.in);
    int n = input.nextInt();
    int[] a = new int[n];
    int[] b = new int[n];

    for (int i = 0; i < n; i++) a[i] = input.nextInt();
    for (int i = 0; i < n; i++) b[i] = input.nextInt();

    Arrays.sort(a);
    Arrays.sort(b);

    long total = 0;
    for (int i = 0; i < n; i++) {
      total += Math.abs((long) a[i] - b[i]);
    }
    System.out.println(total);
  }
}
この記事は著者により CC BY 4.0 ライセンスで公開されています。