BZOJ4371: [IOI2015]sorting排序

我们假设E不操作，A把所有元素复位的最优解是枚举i，若他不在位置i上就和位置i交换，把他转化到图上正确性显然
现在E操作，我们假设位置0~n-1上有碟子0~n-1，碟子i上有苹果i
我们让E操作是交换碟子，A操作是交换苹果
发现这和原问题是等价的，于是我们就可以把E的操作和A的操作分离开来

二分答案，执行完E的前mid次操作后判A是否可以按照上面提到的最优解复原

code：

#include<set>
#include<map>
#include<deque>
#include<queue>
#include<stack>
#include<cmath>
#include<ctime>
#include<bitset>
#include<string>
#include<vector>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<climits>
#include<complex>
#include<iostream>
#include<algorithm>
#define ll long long
using namespace std;

inline void read(int &x)
{
    char c; while(!((c=getchar())>='0'&&c<='9'));
    x=c-'0';
    while((c=getchar())>='0'&&c<='9') (x*=10)+=c-'0';
}
int ch[10]; int cn;
inline void output(int x)
{
    if(!x) { putchar('0');return; }
    while(x) ch[++cn]=x%10,x/=10;
    while(cn) putchar('0'+ch[cn--]);
}
const int maxn = 210000;

int n,m;
int s[maxn],X[maxn],Y[maxn];

int p[maxn],pi[maxn];
int a[maxn],ai[maxn];

int t[maxn][2],tp;
int judge(int mid,int ans)
{
    for(int i=0;i<n;i++) a[i]=s[i];
    for(int i=1;i<=mid;i++) swap(a[X[i]],a[Y[i]]);
    int use=0;
    for(int i=0;i<n;i++) while(i!=a[i]) swap(a[i],a[a[i]]),use++;

    if(ans)
    {
        for(int i=0;i<n;i++) p[i]=i,pi[i]=i;
        for(int i=1;i<=mid;i++)
        {
            int x=pi[X[i]],y=pi[Y[i]];
            swap(p[x],p[y]);
            pi[p[x]]=x,pi[p[y]]=y;
        }

        for(int i=0;i<n;i++) a[i]=s[i],ai[a[i]]=i;
        tp=0;
        for(int i=0;i<n&&tp<=mid;i++) if(p[ai[i]]!=i)
        {
            ++tp,t[tp][0]=i,t[tp][1]=a[pi[i]];
            int x=ai[t[tp][0]],y=ai[t[tp][1]];
            swap(a[x],a[y]);
            ai[a[x]]=x,ai[a[y]]=y;
        }
        while(ans&&tp<mid) ++tp,t[tp][0]=t[tp][1]=0;

        for(int i=0;i<n;i++) p[i]=i,pi[i]=i;
        for(int i=0;i<n;i++) a[i]=s[i],ai[a[i]]=i;
        for(int i=1;i<=mid;i++)
        {
            int x,y;
            x=pi[X[i]],y=pi[Y[i]];
            swap(p[x],p[y]);
            pi[p[x]]=x,pi[p[y]]=y;

            x=ai[t[i][0]],y=ai[t[i][1]];
            t[i][0]=p[x],t[i][1]=p[y];
            swap(a[x],a[y]);
            ai[a[x]]=x,ai[a[y]]=y;
        }
    }

    return use<=mid;
}

int main()
{
    //freopen("tmp.in","r",stdin);
    //freopen("tmp.out","w",stdout);

    read(n);
    for(int i=0;i<n;i++) read(s[i]);
    read(m); m=n-1;
    for(int i=1;i<=m;i++) read(X[i]),read(Y[i]);

    int l=0,r=n-1;
    while(l<=r)
    {
        int mid=(l+r)>>1;
        if(judge(mid,0)) r=mid-1;
        else l=mid+1;
    }
    printf("%d\n",r+1);
    judge(r+1,1);
    for(int i=1;i<=r+1;i++) output(t[i][0]),putchar(' '),output(t[i][1]),putchar('\n');

    return 0;
}