AtCoder Grand Contest 006 E - Rotate 3x3 树状数组

题意

一开始有一个3*n的矩阵，初始时位置(i,j)上是(j-1)*3+i。现在给出一个目标状态，问能否通过每次把一个3*3的矩阵旋转180度来得到该目标状态。
n<=100000

分析

我们可以把问题看成能否由目标状态变成初始状态。
若某一列是正的，则系数为1，否则系数为-1。那么问题就变成了，给你一列数，每次可以选择连续的三个数然后将这三个数翻转并取反，问能否变成1,2,…,n。
先不考虑符号，那么每次就相当于交换奇数位或偶数位上相邻的两个数。不难发现对于奇数位和偶数位我们分别要交换其逆序对数量这么多次。
假设现在已经排好序了，不难发现我们可以通过一些操作来使得两个下标奇偶性相同的位置同时取反，且其他位置不变。
那么答案就只跟逆序对的奇偶性和负号的奇偶性有关。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;

const int N=100005;

int n,a[N],c[N];

int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

int find(int x)
{
    int ans=0;
    while (x) ans^=c[x],x-=x&(-x);
    return ans;
}

void ins(int x)
{
    while (x<=n) c[x]^=1,x+=x&(-x);
}

int main()
{
    n=read();
    for (int i=1;i<=n;i++)
    {
        int x=read();
        if (x%3==2) {puts("No");return 0;}
        if (x%3==1) a[i]=x/3+1;
        else a[i]=-x/3;
    }
    for (int i=1;i<=n;i++)
    {
        int x=read();
        if ((x+2)/3!=abs(a[i])||x%3!=2) {puts("No");return 0;}
    }
    for (int i=1;i<=n;i++)
    {
        int x=read();
        if ((x+2)/3!=abs(a[i])||x%3==2) {puts("No");return 0;}
    }
    for (int i=1;i<=n;i++) if ((i&1)!=(abs(a[i]&1))) {puts("No");return 0;}
    int f1=0,f2=0;
    for (int i=1;i<=n;i+=2)
    {
        if (a[i]<0) f1^=1;
        f2^=find(n)^find(abs(a[i]));
        ins(abs(a[i]));
    }
    memset(c,0,sizeof(c));
    int f3=0,f4=0;
    for (int i=2;i<=n;i+=2)
    {
        if (a[i]<0) f3^=1;
        f4^=find(n)^find(abs(a[i]));
        ins(abs(a[i]));
    }
    if (f1!=f4||f2!=f3) {puts("No");return 0;}
    puts("Yes");
    return 0;
}