给m个数,每个数在1-n之间,有n个排列,求n次排列中,这m个数的相邻两个数之间的位置的绝对值之差。
可以发现,对于第i个排列,其实就是把i 和 i+1 换个位置而已,所以如果有两个相邻的数,不是i,也不是i+1的话,那么这两个数第i次变化的对答案的贡献就是0
所以只需要考虑和i,以及i+1相邻的那些数即可
#include<bits/stdc++.h>
using namespace std;
#define f first
#define s second
typedef long long LL;
typedef pair<int,int> PII;
typedef pair<LL,LL> PLL;
const int N = 2e5 + 10;
int n,m,a[N],pos[N];
map<int,int> lin[N];
int main(){
ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
//freopen("data.in","r",stdin);
//freopen("data.out","w",stdout);
cin>>n>>m;
for(int i = 1 ; i <= n; i ++) pos[i] = i;
for(int i = 0 ;i < m ;i ++) {
cin>>a[i];
}
for(int i = 0 ; i < m - 1;i ++) {
if(a[i] != a[i + 1] ){
if(!lin[a[i]].count(a[i + 1])) lin[a[i]][a[i+1]] = 0 ,lin[a[i+1]][a[i]] = 0;
lin[a[i]][a[i+1]]++ ;
lin[a[i+1]][a[i]]++ ;
}
}
// for(int i = 1; i <= n; i ++) {
// for(auto u : lin[i]) cout<<u.f<<' '<<u.s<<' ';
// cout<<endl;
// }
LL sum = 0 ;
vector<LL> ans;
for(int i = 0 ; i < m - 1; i ++) sum += abs(pos[a[i]] - pos[a[i+1]]);
ans.push_back(sum);
for(int i = 1 ; i < n ; i ++){
//cout<<sum<<' ';
for(auto u : lin[i]) sum -= (LL)(abs(pos[u.f] - pos[i]) ) * u.s;
for(auto u : lin[i + 1]) sum -= (LL)(abs(pos[u.f] - pos[i + 1])) * u.s;
swap(pos[i] , pos[i + 1]);
//cout<<sum<<endl;
for(auto u : lin[i]) sum += (LL)(abs(pos[u.f] - pos[i]) ) * u.s;
for(auto u : lin[i + 1]) sum += (LL)(abs(pos[u.f] - pos[i + 1])) * u.s;
ans.push_back(sum);
}
for(auto u : ans) cout<<u<<' ';cout<<endl;
return 0;
}