tookunn’s diary

主に競技プログラミング関係

Codeforces Educational Round 10 C

考察

区間[i,j](0 \leq i \leq j \lt N)にFoePairsが含まれている場合、区間[i,k](j \lt k)にはForPairsが含まれているので、カウントしない。

・つまり,区間の始点iからFoePairsが含まれているような最小の終点jを分かれば良さそう。

・尺取り法を使うっぽい。

・問題ページのサンプル2のイメージ

9 7 2 3 1 4 6 5 8
[i,j] = [8,8]
9 7 2 3 1 4 6 5 8
[i,j] = [7,8]
9 7 2 3 1 4 6 5 8
[i,j] = [6,8]
9 7 2 3 1 4 6 5 8
[i,j] = [5,6]
9 7 2 3 1 4 6 5 8
[i,j] = [4,5]
9 7 2 3 1 4 6 5 8
[i,j] = [3,5]
9 7 2 3 1 4 6 5 8
[i,j] = [2,5]
9 7 2 3 1 4 6 5 8
[i,j] = [1,1]
9 7 2 3 1 4 6 5 8
[i,j] = [1,1]
9 7 2 3 1 4 6 5 8
以上の図のi,jに対して,j - i + 1をそれぞれ足していくと解になる。
参考:
pekempey.hatenablog.com
mayokoex.hatenablog.com

ソースコード

import java.io.IOException;
import java.io.InputStream;
import java.io.PrintWriter;
import java.util.ArrayList;
import java.util.NoSuchElementException;

public class C {

	int N,M;
	int[] P,position;
	ArrayList<Integer>[] pair;
	public void solve() {

		N = nextInt();
		M = nextInt();
		P = new int[N];
		position = new int[N];
		pair = new ArrayList[N];
		for(int i = 0;i < N;i++){
			pair[i] = new ArrayList<Integer>();
		}

		for(int i = 0;i < N;i++){
			P[i] = nextInt() - 1;
			//P[i]が出現するインデックスを覚えとく
			position[P[i]] = i;
		}

		for(int i = 0;i < M;i++){
			int a = nextInt() - 1;
			int b = nextInt() - 1;

			int x = position[a];
			int y = position[b];


			if(x > y){
				int t = x;
				x = y;
				y = t;
			}
			//x < y
			pair[x].add(y);
		}



		long ans = 0;
		int k = N;
		for(int i = N - 1;i >= 0;i--){

			//i < pair[i].get(j)である。
			//iに一番近いインデックスをkとする
			for(int j = 0;j < pair[i].size();j++){
				k = Math.min(k, pair[i].get(j));
			}
			ans += k - i;
		}
		out.println(ans);

	}

	public static void main(String[] args) {
		out.flush();
		new C().solve();
		out.close();
	}

	/* Input */
	private static final InputStream in = System.in;
	private static final PrintWriter out = new PrintWriter(System.out);
	private final byte[] buffer = new byte[2048];
	private int p = 0;
	private int buflen = 0;

	private boolean hasNextByte() {
		if (p < buflen)
			return true;
		p = 0;
		try {
			buflen = in.read(buffer);
		} catch (IOException e) {
			e.printStackTrace();
		}
		if (buflen <= 0)
			return false;
		return true;
	}

	public boolean hasNext() {
		while (hasNextByte() && !isPrint(buffer[p])) {
			p++;
		}
		return hasNextByte();
	}

	private boolean isPrint(int ch) {
		if (ch >= '!' && ch <= '~')
			return true;
		return false;
	}

	private int nextByte() {
		if (!hasNextByte())
			return -1;
		return buffer[p++];
	}

	public String next() {
		if (!hasNext())
			throw new NoSuchElementException();
		StringBuilder sb = new StringBuilder();
		int b = -1;
		while (isPrint((b = nextByte()))) {
			sb.appendCodePoint(b);
		}
		return sb.toString();
	}

	public int nextInt() {
		return Integer.parseInt(next());
	}

	public long nextLong() {
		return Long.parseLong(next());
	}

	public double nextDouble() {
		return Double.parseDouble(next());
	}
}