P r o b l e m \mathrm{Problem} Problem
工作使艾奇快乐。
勤奋的工作为国家直接贡献了GDP,艾奇认为只要对国家有利,即使牺牲自己生命也心甘情愿,绝不会因为自己可能受到祸害而躲开。
当艾奇无聊的时候,她就会去工作,然而并不是每次工作都是轻松而愉悦的。
当天艾奇又一次来到了学校,等待着她的是一个有n 行m 列的巨大的矩阵和q个任务。对于每个任务,艾奇被要求交换这个矩阵中的两个子矩阵。
每个任务,艾奇会获得六个正整数x1,y1,x2,y2,h,w。x1,y1 代表了第一个矩阵 左上角的行列位置(即在第x1 行第y1 列);x2,y2 代表了第二个矩阵左上角的行列位置,h,w 代表了这两个矩阵的高和宽(即行数和列数)。
数据保证每两个需要交换的矩阵互不相交或相邻。也就是说,没有任何一个元素同时属于这两个矩阵,也不存在某两个元素分别属于两个矩阵且相邻(共边)。
输入第一行三个整数n,m,q,表示矩阵的行数,列数,交换次数
接下来n行,每行m个数,表示原始矩阵
接下来q行,每行6个正整数x1,y1,x2,y2,h,w,表示要交换的两个矩阵的左上角坐标及所求矩阵的行数和列数。
输出n行m个数,表示完成q次交换后的矩阵
S o l u t i o n \mathrm{Solution} Solution
一道非常神奇的题,这道题的解题关键就在于如何解决又换矩阵这一个问题。
一些废话:观察数据范围,我们发现可以用 O ( ( n + m ) q ) O((n+m)q) O((n+m)q)的复杂度解决问题。
考虑链的做法:怎么做矩阵的交换最方便呢,由于矩阵不想交,我们可以维护一个链表,即对 [ l 1 , r 1 ] [l_1,r_1] [l1,r1]和 [ l 2 , r 2 ] [l_2,r_2] [l2,r2]进行交换的话,需要将 r 1 r_1 r1和 r 2 r_2 r2的后继交换。
那么维护到矩阵上同理,我们可以维护一个十字链表,对于每一个节点记录右边的点和下面的节点,这样我们就可以可以做到快速交换了。
那么如何交换呢,这里我们需要在 O ( n + m ) O(n+m) O(n+m)的时间内完成交换。
- 首先找到 ( x 1 , y 1 ) (x_1,y_1) (x1,y1)的编号,再找到 ( x 2 , y 2 ) (x_2,y_2) (x2,y2)的编号。
- 先向右 w w w时,交换下指针;再向下 h h h时,交换右指针。
- 先向下 h h h时,交换右指针;再向右 w w w时,交换下指针。
最后是输出,这个就很简单了,直接遍历链表即可。
C o d e \mathrm{Code} Code
#include <bits/stdc++.h>
using namespace std;
const int N = 1200;
int n, m, k;
struct List {
int val, r, d;
} a[N * N];
int read(void)
{
int s = 0, w = 0; char c = getchar();
while (c < '0' || c > '9') w |= c == '-', c = getchar();
while (c >= '0' && c <= '9') s = s*10+c-48, c = getchar();
return w ? -s : s;
}
inline int C(int x,int y) {
return x * (m + 1) + y;
}
void change(void)
{
int x1 = read(), y1 = read(), x2 = read(), y2 = read();
int h = read(), w = read(), t1 = 0, t2 = 0;
for (int i=1;i<x1;++i) t1 = a[t1].d;
for (int i=1;i<y1;++i) t1 = a[t1].r;
for (int i=1;i<x2;++i) t2 = a[t2].d;
for (int i=1;i<y2;++i) t2 = a[t2].r;
int p1 = t1, p2 = t2;
for (int i=1;i<=w;++i) {
p1 = a[p1].r, p2 = a[p2].r;
swap(a[p1].d,a[p2].d);
}
for (int i=1;i<=h;++i) {
p1 = a[p1].d, p2 = a[p2].d;
swap(a[p1].r,a[p2].r);
}
p1 = t1, p2 = t2;
for (int i=1;i<=h;++i) {
p1 = a[p1].d, p2 = a[p2].d;
swap(a[p1].r,a[p2].r);
}
for (int i=1;i<=w;++i) {
p1 = a[p1].r, p2 = a[p2].r;
swap(a[p1].d,a[p2].d);
}
return;
}
void output(void)
{
int t = 0;
for (int i=1;i<=n;++i)
{
t = a[t].d;
int p = t;
for (int i=1;i<=m;++i) {
p = a[p].r;
printf("%d ", a[p].val);
}
puts("");
}
return;
}
int main(void)
{
n = read(), m = read(), k = read();
for (int i=1;i<=n;++i)
for (int j=1;j<=m;++j)
a[C(i,j)].val = read();
for (int i=0;i<=n;++i)
for (int j=0;j<=m;++j) {
a[C(i,j)].r = C(i,j+1);
a[C(i,j)].d = C(i+1,j);
}
while (k --) change();
output();
return 0;
}