Codeforces Round #324 (Div. 2) E. Anton and Ira 贪心

 

题目

 

题意：

　　    给出两个有相同元素的数组a，数组b，  使 a[i]和 a[j] 进行交换，得到数组 b ， 每次交换的代价为abs( i - j )。

　　　 要求输出最小的总代价，交换的步数，以及每次交换的两个位置 。

 

思路：

　　    如果    数组 a :  5 2 1 4 6 3

　　　　　     数组 b:   1 2 3 4 5 6   

　　　 只需对数组b的每项b[ i ]， 尽量使数组a中值为b[ i ]的位置 和 别的a[ j ]所在的位置  交换，使得最终a[i]==b[i] 。

   　　  b[1]=1，在a中找到 a[3]=1，位置为3 ，记做k=3， 此时a中 5占据了1的位置，尝试着交换看看。 5 在b中的位置是第5，5>=k， 说明如果a中 a[1]和a[3]交换，即 值 5 和1交换，就得到 1 2 5 4 6 3， 这样1就归位了（a[i]==b[i]），而且5此时的位置变成3，离5更近了。  这次交换，使得1归位，使5离目的地更近了，说明在交换过程中代价是最小的，没有交换过头的现象，这样的交换是最贪心的。

　　　  b[1]处理完了，再处理b[2]，按上面的思路使a[2]==b[2]， 以此类推。  应该是O(n^2)的复杂度。

 

#include<iostream>
#include<cstdio>
#include <cctype>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<string>
#include<cmath>
#include<set>
#include<vector>
#include<stack>
#include<queue>
#include<map>
using namespace std;
#define ll long long
#define mem(a,x) memset(a,x,sizeof(a))
#define se second
#define fi first
const ll mod=1e9+7;
const int INF= 0x3f3f3f3f;
const int N=2e6+5;

int n,cnt,ans;
int a[N],b[N];
int p[N];
int ans1[N],ans2[N];


int main()
{
    
        scanf("%d",&n);
        for(int i=1;i<=n;i++)    scanf("%d",&a[i]);
        for(int i=1;i<=n;i++)    scanf("%d",&b[i]),p[b[i]]=i;
        
        cnt=0;
        ans=0;
        for(int i=1;i<=n;i++)
        {
            int k=i;
            
            if(b[i]!=a[i])  //找到b[i]在a中的位置k 
            {
                for(int j=i+1;j<=n;j++)
                {
                    if(b[i]==a[j]){
                        k=j; break;  
                    }
                }
            }
            
            while(k!=i) //意味着还不归位，a[i]!=b[i] 
            {
                for(int j=i;j<k;j++)
                {
                    if(p[a[j]]>=k) //a[j]的位置在k的后面
                    {
                        ans+=abs(k-j);
                        ans1[cnt]=j;
                        ans2[cnt]=k;
                        cnt++;
                        
                        swap(a[j],a[k]);
                        swap(j,k); //这一步 保证k随着交换移前，也保证此for的终结 
                        
                        
                    } 
                }
            }
            
        }
        
        cout<<ans<<endl;
        cout<<cnt<<endl;
        for(int i=0;i<cnt;i++)
            cout<<ans1[i]<<' '<<ans2[i]<<endl;
        
    
}

 

 

　　如果n太大，想直接得到步数

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int a[200005],b[200005];
int conv[200005];
int main()
{
    int zu;
    scanf("%d",&zu);
    while(zu--)
    {
        int n;
        scanf("%d",&n);
        for(int i=1;i<=n;i++) scanf("%d",&a[i]),conv[a[i]]=i;
        for(int i=1;i<=n;i++) scanf("%d",&b[i]);
        ll ans=0;
        for(int i=1;i<=n;i++) ans+=abs(i-conv[b[i]]);
        printf("%lld\n",ans/2);
    }
}