捡石头

题目描述

地上有2N个石头，排成了一条线，相邻的石头距离为1，石头之间有着不同的大小，有N种大小不同 的石头，即相同大小的石头有2个，现将石头按照从小到大的顺序依次编号为1到N，有2个石头共享 相同的编号，现在小武和小林要同时从最左边的石头出发，按照石头大小依次捡起编号为1到N的石 头，并且相同编号的石头同一个人只能捡起来一次，现在他们想把地上的石头都捡完，求两个人的行 走的最短距离和为多少？

输入格式

第一行一个正整数N 第二行2N个数，按照石头从左到右的顺序依次给出石头的编号

输出格式

一行一个数表示行走的最短距离和

输入输出样例

输入 #1

3
1 1 2 2 3 3

输出 #1

9

输入 #2

4
2 2 3 4 4 1 1 3

输出 #2

33

输入 #3

3
1 3 1 2 3 2

输出 #3

11

说明/提示

数据范围

对于16%的数据，N<=10
对于36%的数据，N<=1000
对于60%的数据，N<=10000
对于100%的数据，N<=100000

样例解释

对于样例1
第一个人捡第1,3,5的石头，大小分别为1,2,3，行走距离为4
第二个人捡第2,4,6的石头，大小分别为1,2,3，行走距离为5
总距离为9

---

解题思路

可以发现每两个点之间的行走可能只有两种

• （偏左起点 -> 偏左终点 + 偏右起点 -> 偏右终点）
• （偏左起点 -> 偏右终点 + 偏右起点 -> 偏左终点）

所以只需要将两种情况中短的情况加入答案就好啦

---

Code

#include <iostream>
#include <cstdio>
#include <cmath>

using namespace std;

int n, x;
long long a[100100][2], ans, xx, yy;

int main(){
	scanf ("%d", &n);
	for (int i = 1; i <= n * 2; i++)
	{
	     scanf ("%d", &x);
	     if (a[x][0])
	         a[x][1] = i;
	        else a[x][0] = i;
	}
	ans = a[1][0] + a[1][1] - 2;//到石头1的距离是唯一的，而且因为从第一块石头出发路程少1，两个石头1路程少2
	for (int i = 2; i <= n; i++)
	{
		xx = abs (a[i][0] - a[i - 1][0]) + abs (a[i][1] - a[i - 1][1]);//两种可能
		yy = abs (a[i][0] - a[i - 1][1]) + abs (a[i][1] - a[i - 1][0]);
		ans += min (xx, yy);
	}
	printf ("%lld", ans);
}