HDU-6048 Puzzle（思维题）

传送门：HDU-6048

又是一道结论题（吐槽：这次多校好多结论题啊）

根据九宫格问题的结论：将矩阵中的数按从上到下从左到右排成一列，其逆序对如果为偶数则一定有解，否则一定无解

求逆序对的方法：

每一轮要在数列中从第一个数开始每隔P个选一个数出来，因此选出来的数对逆序对的贡献为：0，P-1，2(P-1)，3(P-1)...构成一个等差数列。选完一轮之后要将剩下的从头再选

#include<bits/stdc++.h>
using namespace std;
#define mem(x,y) memset(x,y,sizeof x)
typedef long long LL;
typedef pair<int, int> PII;
const int inf = 0x3f3f3f3f;
const LL INF = 0x3f3f3f3f3f3f3f3f;
const LL mod = 1e9 + 7;

int main() {
int T,n,m,p;
//freopen("in.txt", "r", stdin);
scanf("%d",&T);
while(T--){
scanf("%d%d%d",&n,&m,&p);
int cnt=n*m-1,ans=0;
while(cnt>p){
int c=(cnt-1)/p;
cnt-=c+1;
ans+=c*(c*(p-1)+p-1)/2;
}
printf("%s\n",ans%2?"NO":"YES");
}
return 0;
}