B.Two Cakes

Posted 2021-02-11 ydddd

链接：https://codeforces.com/contest/1130/problem/B

题意：

给定n和 2 * n个数，表示i位置卖ai层蛋糕，

有两个人在1号，必须严格按照1-n的顺序买蛋糕，同时每个店只买一个蛋糕 。

求所需的最短时间。

思路：

将每种蛋糕对应位置记录在二维数组。

从1-n挨个买，根据上一次的位置算出消耗。

代码：

#include <bits/stdc++.h>
using namespace std;

typedef long long LL;

const int MAXN = 1e5 + 10;

int a[MAXN][2];

int main()
{
    int n, x;
    cin >> n;
    for (int i = 1;i <= n * 2;i++)
    {
        cin >> x;
        if (a[x][0])
            a[x][1] = i;
        else
            a[x][0] = i;
    }
    int p1 = 1, p2 = 1;
    LL res = 0;
    for (int i = 1;i <= n;i++)
    {
        res += abs(a[i][0] - p1) + abs(a[i][1] - p2);
        p1 = a[i][0];
        p2 = a[i][1];
    }
    cout << res << endl;

    return 0;
}