B. Two Cakes
题目大意
输入
2
∗
n
2*n
2∗n个数,表示编号为
a
i
ai
ai的蛋糕在第i个位置(
1
−
n
1-n
1−n每个数出现两次),现在有两个人在第一个位置,他们两个人要分别要拿n个蛋糕,且每个人只能按
1
−
n
1-n
1−n的顺序拿蛋糕,问两个人最少走多远才能拿完蛋糕
解题思路
- 我们可以用二维数组记录一下每个蛋糕的位置后,从 1 − n 1-n 1−n开始遍历小蛋糕的位置
- 然后贪心的判断从上一个位置到当前位置怎么走路程最短,用 a n s ans ans记录一下最短路程
- 记得两人原始是在
1
1
1位置,所以最后要加入两个
1
代表初始位置 - 具体操作见代码
附上代码
#pragma GCC optimize("-Ofast","-funroll-all-loops")
#include<bits/stdc++.h>
#define int long long
#define lowbit(x) (x &(-x))
#define endl '\n'
using namespace std;
const int INF=0x3f3f3f3f;
const int dir[4][2]={-1,0,1,0,0,-1,0,1};
const double PI=acos(-1.0);
const double e=exp(1.0);
const double eps=1e-10;
const int M=1e9+7;
const int N=2e5+10;
typedef long long ll;
typedef pair<int,int> PII;
typedef unsigned long long ull;
vector<int> v[N];
signed main(){
ios::sync_with_stdio(false);
cin.tie(0);cout.tie(0);
int n,ans=0;
cin>>n;
for(int i=1;i<=2*n;i++){
int t;
cin>>t;
v[t].push_back(i);
}
v[0].push_back(1);
v[0].push_back(1);
for(int i=1;i<=n;i++){
int a=abs(v[i][0]-v[i-1][0])+abs(v[i][1]-v[i-1][1]);
int b=abs(v[i][1]-v[i-1][0])+abs(v[i][0]-v[i-1][1]);
ans+=min(a,b);
}
cout<<ans<<endl;
return 0;
}