about summary refs log tree commit diff
path: root/2ndary/09/TP-HN-2014/cau4.c
blob: 49e3df6300c31282c3666dbe178b24f218ebf081 (plain) (blame)
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
#include <stdlib.h>
#include <stdio.h>

int cmp(const void *x, const void *y)
{
	return *(int *) x - *(int *) y;
}

int main()
{
	FILE *f = fopen("CAU4.INP", "r");
	int n, d, k;
	fscanf(f, "%d %d", &n, &d);
	int *a = (int *) malloc(n * sizeof(int));
	for (k = 0; k < n; k++)
		fscanf(f, "%d", a + k);
	k = a[--d];
	fclose(f);

	qsort(a, n, sizeof(int), cmp);
	int t = 0;
	for (int i = 1; i < n; i++)
		t += abs(a[i] - a[i - 1]);
	int idx = (int *) bsearch(&k, a, n, sizeof(int), cmp) - a;
	if ((idx - d) * (idx - n + d + 1)) {
		int t0, t1;
		if (idx < d) {
			t0 = (t - abs(a[n - d + idx] - a[n - d + idx - 1])
			      + abs(a[n - d + idx] - *a));
			d = n - d - 1;
			t1 = (t - abs(a[idx - d] - a[idx - d - 1])
			      + abs(a[n - 1] - a[idx - d - 1]));
		} else {
			t0 = (t - abs(a[idx - d] - a[idx - d - 1])
			      + abs(a[n - 1] - a[idx - d - 1]));
			d = n - d - 1;
			t1 = (t - abs(a[n - d + idx] - a[n - d + idx - 1])
			      + abs(a[n - d + idx] - *a));
		}
		t = (t0 < t1) ? t0 : t1;
	}

	f = fopen("CAU4.OUT", "w");
	fprintf(f, "%d\n", t);
	fclose(f);

	return 0;
}