USACO 2020~2021 February Contest GOLD 题解(3)

USACO 2020~2021 二月黄金组 题解(3)

3. Count The Cows

As is typical, Farmer John’s cows have spread themselves out along his largest pasture, which can be regarded as a large 2D grid of square “cells” (picture a huge chessboard).
The pattern of cows across the pasture is quite fascinating. For every cell (x,y) with x≥0 and y≥0, there exists a cow at (x,y) if for all integers k≥0, the remainders when ⌊x3k⌋ and ⌊y3k⌋ are divided by three have the same parity. In other words, both of these remainders are odd (equal to 1), or both of them are even (equal to 0 or 2).
FJ is curious how many cows are present in certain regions of his pasture. He asks Q queries, each consisting of three integers xi,yi,di. For each query, FJ wants to know how many cows lie in the cells along the diagonal range from (xi,yi) to (xi+di,yi+di) (endpoints inclusive).

题目大意：农夫John的牛在一个二维方阵里吃草。对于方阵中的某个格子(x,y)（x,y为自然数），该方格中有牛当且仅当对于所有非负整数k，都有$\lfloor x/3^k\rfloor$，$\lfloor y/3^k\rfloor$除以3的余数同奇偶。现在John一共会作出Q次询问（1$\le$Q$\le$ 10⁴ ），每个询问包含三个整数x,y,d(0$\le$x,y,d$\le$ 10¹⁸)，询问在以(x,y)上顶点，(x+d,y+d)为下顶点的对角线上牛的总头数。

思路分析： 这道题乍一看是一道描述不算简单的数论题，我们先打一下表（其实通过题目中给出的样例也可以看出一些规律）：

101000101
010000010
101000101
000101000
000010000
000101000
101000101
010000010
101000101

我们有如下猜想：整个方格是一个形状为“X”的分形图。下面提供对这一猜想的粗略证明：考虑背靠原点3*3个边长为3^k的方格，将每个分割为9个边长为3^k-1的小方格。可以发现，每个大方格的横纵坐标在除以3取余数之后，都会被映射至左上方大方格内的相对位置一致的小方格：如果大方格横纵坐标除以三之余数奇偶性不同，那么这个大方格不满足条件，因此对应小方格也不满足条件。运用数学归纳法的思想，可以推出对于任意的正整数k,在背靠原点的3 * 3个边长为3^k的方格，横纵坐标奇偶性不同的方格没有牛，而横纵坐标奇偶性相同的方格与左上方小方格构图相同。因此，整个方格是一个形状为“X”的分形图。

有了这个重要的观察，我们可以运用递归的思想，将边长为3^k的正方形对角线在O(1)的时间内转化为对边长为3^k-1的正方形对角线的求解。具体来说,定义f(k,dif)为在以原点为起点，边长为3^k的正方形中所有满足x−y=dif且有牛的方格(x,y)的个数（也就是该正方形左下角边长为dif的正方形的对角线上有牛的方格的个数）。那么：
$$f(k,dif)=\{\begin{array}{ll}3f(k-1,dif)& \text{dif < 3k−1}\\ f(k-1,dif-2(3^{k-1}))& \text{dif≥ 3k−1}\end{array}$$
回答每个询问时，可以用边长为(x+d)的答案减去边长为(x-1)的答案。算法复杂度为O(Qlog(max(x+d)))。具体解析详见代码。

官方解法代码如下：

#include <bits/stdc++.h>
using namespace std;
#define LL long long
 
vector<LL> po3 = [](){
	vector<LL> res{1};
	for (int i = 1; i < 40; ++i) 
		res.push_back(3*res.back());
	return res;
}();//将3的k次幂先预处理，方便使用

LL full[40];
void gen_full(int k, ll dif) //预处理x-y=dif的正方形的函数
{ 
	dif = abs(dif);//dif为负时横纵颠倒，不影响
	if (k == 0) 
	{
		full[k] = (dif == 0);//边界处理：横纵坐标相等时显然有牛
		return;
	}
	if (dif >= po3[k-1])
	{
		gen_full(k-1,dif-2*po3[k-1]);
		full[k] = full[k-1];
	} 
	else
	{
		gen_full(k-1,dif);
		full[k] = 3*full[k-1];
	}//代入转移方程
}
 
ll rec(ll x, ll y, int k) {
	x %= po3[k], y %= po3[k];//求余数
	if (k == 0) return 1;//处理边界情况
	if (x < y) swap(x,y);
	//考虑“X”的形状，9个位置分类讨论
	if (x-y >= po3[k-1]) {
		if (x < 2*po3[k-1]) return 0;
		if (y < po3[k-1]) return rec(x,y,k-1);
		if (y >= po3[k-1]) return full[k-1];
	}
	if (x < po3[k-1]) return rec(x,y,k-1);
	if (y < po3[k-1]) return full[k-1];
	if (x < 2*po3[k-1]) return full[k-1]+rec(x,y,k-1);
	if (y < 2*po3[k-1]) return 2*full[k-1];
	return 2*full[k-1]+rec(x,y,k-1);
}
LL diag(ll x, ll y) {
	if (x < 0 || y < 0) return 0;
	gen_full(39,x-y);//预处理dif=x-y的最大正方形
	return rec(x,y,39);
}
int Q; 
int main() {
	scanf("%d",&Q);
	while(Q--)
	{
		LL d,x,y;
		scanf("%lld%lld%lld",&d,&x,&y);
		printf("%lld",diag(x+d,y+d)-diag(x-1,y-1));//差分思想
	}
}

本次比赛其他题目：

USACO 2020~2021 二月黄金组 题解（1）Stone Game
USACO 2020~2021 二月黄金组 题解（2）Modern Art 3