#include <stdio.h> static int n, k; static int t[500000]; static void dump(void) { int i; for (i = 0; i < n; i++) fprintf(stderr, " %d", t[i]); putc('\n', stderr); } int main(void) { int i, j; long long r = 0; static int s[500001]; scanf("%d%d", &n, &k); for (i = 0; i < n; i++) scanf("%d", &t[i]); //dump(); for (i = 0, j = k; i < k; i++) { int b = t[i]; if (s[b]) { int b2; while (j < n) { b2 = t[j]; if (s[b2]) j++; else break; } if (j >= n) { r = -1; break; } r += j - i; t[j++] = b; t[i] = b = b2; //fprintf(stderr, "r=%lld i=%d j=%d:", r, i, j); //dump(); } s[b]++; } //fprintf(stderr, "r=%lld i=%d j=%d:", r, i, j); //dump(); printf("%lld\n", r); return 0; }
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 | #include <stdio.h> static int n, k; static int t[500000]; static void dump(void) { int i; for (i = 0; i < n; i++) fprintf(stderr, " %d", t[i]); putc('\n', stderr); } int main(void) { int i, j; long long r = 0; static int s[500001]; scanf("%d%d", &n, &k); for (i = 0; i < n; i++) scanf("%d", &t[i]); //dump(); for (i = 0, j = k; i < k; i++) { int b = t[i]; if (s[b]) { int b2; while (j < n) { b2 = t[j]; if (s[b2]) j++; else break; } if (j >= n) { r = -1; break; } r += j - i; t[j++] = b; t[i] = b = b2; //fprintf(stderr, "r=%lld i=%d j=%d:", r, i, j); //dump(); } s[b]++; } //fprintf(stderr, "r=%lld i=%d j=%d:", r, i, j); //dump(); printf("%lld\n", r); return 0; } |