题目链接:https://codeforc.es/contest/1279/problem/C
Santa has to send presents to the kids. He has a large stack of n presents, numbered from 1 to n; the topmost present has number a1, the next present is a2, and so on; the bottom present has number an. All numbers are distinct.
Santa has a list of m distinct presents he has to send: b1, b2, …, bm. He will send them in the order they appear in the list.
To send a present, Santa has to find it in the stack by removing all presents above it, taking this present and returning all removed presents on top of the stack. So, if there are k presents above the present Santa wants to send, it takes him 2k+1 seconds to do it. Fortunately, Santa can speed the whole process up — when he returns the presents to the stack, he may reorder them as he wishes (only those which were above the present he wanted to take; the presents below cannot be affected in any way).
What is the minimum time required to send all of the presents, provided that Santa knows the whole list of presents he has to send and reorders the presents optimally? Santa cannot change the order of presents or interact with the stack of presents in any other way.
Your program has to answer t different test cases.
Input
The first line contains one integer t (1≤t≤100) — the number of test cases.
Then the test cases follow, each represented by three lines.
The first line contains two integers n and m (1≤m≤n≤105) — the number of presents in the stack and the number of presents Santa wants to send, respectively.
The second line contains n integers a1, a2, …, an (1≤ai≤n, all ai are unique) — the order of presents in the stack.
The third line contains m integers b1, b2, …, bm (1≤bi≤n, all bi are unique) — the ordered list of presents Santa has to send.
The sum of n over all test cases does not exceed 105.
Output
For each test case print one integer — the minimum number of seconds which Santa has to spend sending presents, if he reorders the presents optimally each time he returns them into the stack.
Example
input
2
3 3
3 1 2
3 2 1
7 2
2 1 7 3 4 5 6
3 1
output
5
8
题意
给出一个有 n 个元素的栈 a ,以及 m 个相关元素的出栈的顺序,一个需要的元素出栈后,其他无关的出栈元素可以以任何你需要的顺序再放进去(不能不放回去),最少需要执行几次操作。
分析
我们发现其中一个 bi 是栈中第 pos[bi] 的位置,如果 bi+1 bi+2 … bx 的位置都在 pos[bi] 前面,那么可以在取出 bi 后讲他们按顺序放回去,这样就不用再取出无用的元素了。
代码
#include<bits/stdc++.h>
using namespace std;
int t;
int n,m;
map<int,int> pos;
int b[100007];
long long ans;
int main()
{
scanf("%d",&t);
while(t--)
{
ans = 0;
pos.clear();
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
{
int x;
scanf("%d",&x);
pos[x] = i;
}
for(int i=1;i<=m;i++)
scanf("%d",&b[i]);
int i = 1;
int gone = 0;
while(i <= m)
{
int p = pos[b[i]];
int j = i;
while(j + 1 <= m && pos[b[j + 1]] < p) j++;
int temp = j - i;
ans += (p - 1 - gone) * 2 + 1;
ans += temp;
gone += temp + 1;
i = j + 1;
}
printf("%lld\n",ans);
}
return 0;
}