【JZOJ A组】图书管理员

Description

在他的图书馆里，图书管理员Juric有N个书架，每个书架上能容纳M本书。因为有些书被人们错误放置了，作为一个有强迫症的管理员，Juric想对这些图书做出调动，使它们都放置在它们应该在位置上。他会以如下的方式来移动图书：
1.他可以将一本书移动到相邻的位置（左边一位或右边一位），但仅当那个位置是空的。
2.他可以把一本书拿出来放在手中，然后放置在任意一个书架的任意的空位置上。
当Juric手上拿着一本书时，他不可以移动图书；他手上也不能同时拿多于一本书。
由于昨天把一整套维基百科从一楼搬到了二楼，身心俱疲的Juric希望在移动图书的过程中尽量减少拿出书本的次数。给出初始的图书放置状态与他期望的状态，Juric在过程中最少需要拿出多少次书？

Input

第一行包含两个整数N(1<=N<=1000)与M(1<=M<=1000)。
接下来N行，每一行包含M个整数，在第i行中表示的第i个书架的当前的状态。
数字0表示该位置是空的，而一个非0数字则表示在该位置上摆放的书的编号。所有书从1到K编号，K是书架上书的总数量。
接下来N行，类似地，表示Juric想要的书架的状态。
初始状态与目标状态出现的书的集合是一样的。

Output

一行一个整数，代表最小的拿出书的次数。如果不能成功地移动，输出-1。

Sample Input

输入1：
2 4
1 0 2 0
3 5 4 0
2 1 0 0
3 0 4 5
输入2：
3 3
1 2 3
4 5 6
7 8 0
4 2 3
6 5 1
0 7 8
输入3：
2 2
1 2
3 4
2 3
4 1

Sample Output

输出1：
2
样例1解释：
对于样例1，Jurica把1号书向右移1位，然后拿起2号书并移到第一个书架的第一个空位上，拿起5号书放到第二个书架的第四个空位上。

输出2：
4
输出3：
-1

Data Constraint

对于50%的数据，满足每一本书在当前状态与目标状态出现在同一个书架上。
对于100%的数据，满足1<=N,M<=1000。

思路

一开始总是想一些奇奇怪怪的数据结构，然后发现没那么难。
这题分为两步：

1. 把不是这一行的挪到相应的一行
2. 同一行内的移动

第一步很容易处理，注意一下移动的位置
第二步DP一下就行了

当然两种特殊情况的判断是很显然的

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
#define fe first
#define se second

using namespace std;

const int maxn=1000005,maxm=1005;

typedef pair<int,int> Pi;

Pi Goal[maxn];
bool emp,empty[maxm];
int Refer[maxn],V[maxm][maxm],P[maxm],Vt[maxm],Fa[maxm],Size[maxm],Uni[maxm],Val[maxn],n,m,K,tot;

int MAX(int p)
{
	int tmp=0;
	for(; p; p -= p&-p) tmp=max(tmp,Vt[p]);
	return tmp;
}

void Update(int p,int v)
{
	for(; p<=m; p+=p&-p) Vt[p]=max(Vt[p],v);
}

int Get(int a)
{
	return Fa[a]==a?a: Fa[a]=Get(Fa[a]);
}

void Union(int a,int b)
{
	a=Get(a),b=Get(b);
	if (a==b) return;
	Size[b]+=Size[a];Uni[b]+=Uni[a];Val[b]+=Val[a];
	Fa[a]=b;
}

int main()
{
	freopen("librarian.in","r",stdin),freopen("librarian.out","w",stdout);
	scanf("%d%d", &n, &m);
	for(int i=1; i<=n; i ++) Fa[i]=i,Uni[i]=1;
	for(int i=1; i<=n; i ++)
		for(int j=1; j<=m; j ++)
		{
			scanf("%d", &V[i][j]);
			if (V[i][j]) K ++,Size[i] ++; else
				empty[i]=1,emp=1;
		}
	for(int i=1; i<=n; i ++)
		for(int j=1; j<=m; j ++)
		{
			int t;
			scanf("%d", &t);
			if (t) Goal[t]=Pi(i,j);
		}
	bool diff=0;
	for(int i=1; i<=n; i ++)
		for(int j=1; j<=m; j ++)
			if (V[i][j] && (Goal[V[i][j]].fe != i || Goal[V[i][j]].se != j)) 
				diff=1;
	if (diff && !emp) {printf("-1\n"); return 0;}
	if (!diff) {printf("0\n"); return 0;}
	for(int i=1; i<=n; i ++)
	{
		int tot=0;
		for(int j=1; j<=m; j ++)
			if (Goal[V[i][j]].fe==i) P[++ tot]=Goal[V[i][j]].se;
		for(int j=0; j<=m; j ++) Vt[j]=0;
		for(int j=1; j<=tot; j ++)
		{
			int cr=MAX(P[j]);
			Val[i]=max(Val[i],cr+1);
			Update(P[j],cr+1);
		}
	}
	for(int i=1; i<=n; i ++)
		for(int j=1; j<=m; j ++)
			if (V[i][j]) Union(i,Goal[V[i][j]].fe);
	int ans=0;
	for(int i=1; i<=n; i ++)
		if (Get(i)==i && Size[i] != Val[i])
			ans+=Size[i]-Val[i]+(Size[i]==Uni[i]*m);
	printf("%d\n", ans);
	return 0;
}