题目链接:CodeForces - 221E
题意:
给两个1~n的排列a、b,当a[i]==b[j]时,距离d=|i-j|,每次将序列b左移一位,开头第一个数移到最后一个,移动n-1次,求每次移动后的距离最小值。
思路:
刚开始想的肯定是暴力(现在想干什么都是暴力,一暴力就超时……),然后参考网上的参考答案,发现思路是这样的:
如果b[j]在a[i]之前,每次左移两点之间的距离都会加1;如果b[j]在a[i]之后,每次左移两点之间的距离都会减少1。用一个multiset来储存距离,用i来表示旋转之后距离的变化
代码。
#include<iostream>
#include<set>
#include<algorithm>
using namespace std;
multiset<int>s;//距离可以重复
multiset<int>::iterator pos;
int a[100001],ans,b[100001];
int main()
{
ios::sync_with_stdio(false);
int n,x;
cin>>n;
for(int i=1;i<=n;i++)
{
cin>>x;
a[x]=i;//记录这个数值的位置
}
for(int i=1;i<=n;i++)
{
cin>>b[i];//可以旋转的第二个排列
s.insert(i-a[b[i]]);//s中插入距离,b[i]的编号-a中b[i]这个数字的编号,结果是距离,b[i]在a中这个数值的前面就是负的,在后面就是正的,每一次旋转负的会=1.正的会减一
}
for(int i=0;i<n;i++)
{
pos=s.lower_bound(i);//找到第一个大于或者等于i的第一个数,每旋转一次第一个i就会被废除,i一直变大,象征移动了之后的距离变化
ans=1e5+10;
//已经排序,最小值一定是他俩之中的一个,因为最小的距离是0,移动之后会减去i的
if(pos!=s.end()) ans=min(ans,*pos-i);//正数中最小值与ans中最小的一个
if(pos!=s.begin()) ans=min(ans,i-(*--pos));//负数中最大的数的绝对值(绝对值也是最小值)与ans中最小的一个
cout<<ans<<endl;
x=b[i+1];//x是第i+1个数(旋转前的第一个)
s.erase(s.find(i+1-a[x]));//原来的距离删去
s.insert(i+1-a[x]+n);//改成了它变成最后一个数之后到那个位置的距离。
}
}