취미로 코딩하기

※ 글쓴이는 취미로 코딩을 익혀보는 사람이라 정확하지 않은 내용을 담고 있을 수 있다 ※

이번에 볼 문제는 백준 3013번 문제인 부분 수열의 중앙값이다.
문제는 아래 링크를 확인하자.

https://www.acmicpc.net/problem/3013 

이 문제에서 부분수열이란 연속된 항들로 구성된 부분수열을 의미한다.

주어진 수열의 모든 수는 서로 다르므로, 홀수 길이의 부분수열들 가운데 중앙값이 B인 부분수열의 개수는 다음과 같은 관찰을 통해 계산할 수 있다.

1) 이 부분수열은 B보다 큰 수와 B보다 작은 수의 개수가 같은 개수로 구성되어야한다.

2) 부분수열은 B를 포함해야하므로 (시작~B의 앞에 적힌 숫자), B, (B의 뒤에 적힌 숫자) 와 같은 구성을 가진다.

아래는 제출한 소스코드이다.

#include <iostream>
using namespace std;
typedef long long ll;

int arr[100000];
int diff[200001];

int main() {
	ios::sync_with_stdio(0);
	cin.tie(0);

	ll ans = 0;
	int N, B; cin >> N >> B;
	int idx;
	for (int i = 0; i < N; i++) {
		cin >> arr[i];
		if (arr[i] == B) idx = i;
	}
	
	int cnt = 1;
	
	for (int i = idx; i < N; i++) {
		if (arr[i] > B) cnt++;
		else cnt--;
		diff[100000 + cnt]++;
	}

	cnt = -1;
	
	for (int i = idx; i >= 0; i--) {
		if (arr[i] > B) cnt--;
		else cnt++;
		ans += diff[100000 + cnt];
	}

	cout << ans;
}