Codeforces 1234E

给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;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值