一、题目大意
围棋棋盘,如果某个坐标上下左右的四个方向都存在棋子,那么ans+1,根据输入的棋子数量,求出ans的数量。
二、解题思路
题目中有说到如果程序不会结束,那么输出-1,这其实是无源之水,根本不会发生。
我们可以一列一列的循环,然后针对列建立一个树状数组(线段树也行,树状数组更快)
坐标比较大,需要离散化(离散化就是把有效坐标排好序去重放在数组里,然后用原坐标对应数字再数组元素的顺序来替换掉原坐标的算法,可以参阅《挑战程序设计》第三章-常用技巧精选,或者可以参考鄙人AOJ0531的拙作题解)本题目每个输入的棋子x和y是有效坐标,其余坐标均无效,因为没有棋子的行或列一定无法让ans+1。
之后根据列来排序,列一样的,就根据行来排序(pair默认的就行,first列,second行)
然后记录下每一行的最后一个棋子的坐标(可以定一个数组,初值设置1或0,循环一次所有的棋子,更新到每一行的最大列即可)
然后,同时记录一个bool型的标记数组,来代表某一行是否前面已经有个棋子,如下图
循环每一列的时候,把当前元素和当前列上一个元素之间的元素集体+1(树状数组操作)update(上一个元素的列+1,当前元素列-1,1)这里需要判断下上一个的列+1和当前列-1的大小,如果大于等于那就不要更新了
同时遇到每一行第一个棋子时,要把这一行标记上,然后这一行的位置更新到0(更新到0是因为这一行之前左边没有棋子,如果左边没有棋子,那么这些+1的情况,即便上下有子也不应该记录到答案里,为的就是防止下图中红色箭头的位置被错误记录了),这样下次再碰到这一行的棋子,就可以代表两者之间的部分位置可以加到答案里。
然后更新到每一行最后一列的时候(这里可以通过之前记录的行最大列的数组来判断是不是最后一列),如果这一行之前没有被标记过,即这一行的最后一个棋子左边没有棋子,那么这一行+1的那些坐标不算数,上下有子,右边也有,但是左边没有那就不行,直接continue。
如果这一行标记过,那那表左边有棋子,同时循环到的这一行的最后一个棋子是它右边的,然后更新树状数组时的区间边界是它上下的,那么树状数组求出这一行的数量,要加到ans里,我这个思路就是如下图所示,每一列右边的一些数子,就是遇到走过某一列的时候树状数组渲染到正常的样子(非树形求和的那种),然后红色的箭头的就代表走到某一行最后一列了增加ans,
表达的不清晰,见谅!
三、代码
#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
typedef long long ll;
typedef pair<int, int> P;
P num[100010];
ll bit0[131080], bit1[131080], ans;
int x[100010], y[100010], xLen, yLen, n, n_, maxCol[100010];
bool activeRow[100010];
void input()
{
for (int i = 1; i <= n_; i++)
{
scanf("%d%d", &num[i].first, &num[i].second);
x[i] = num[i].first;
y[i] = num[i].second;
activeRow[i] = false;
maxCol[i] = 1;
}
sort(x + 1, x + (1 + n_));
sort(y + 1, y + (1 + n_));
}
void compress()
{
xLen = 1;
yLen = 1;
for (int i = 2; i <= n_; i++)
{
if (x[xLen] != x[i])
{
x[++xLen] = x[i];
}
if (y[yLen] != y[i])
{
y[++yLen] = y[i];
}
}
for (int i = 1; i <= n_; i++)
{
num[i].first = lower_bound(x + 1, x + (xLen + 1), num[i].first) - x;
num[i].second = lower_bound(y + 1, y + (yLen + 1), num[i].second) - y;
if (maxCol[num[i].second] < num[i].first)
{
maxCol[num[i].second] = num[i].first;
}
}
}
void init()
{
n = 131072;
for (int i = 0; i <= n; i++)
{
bit0[i] = 0LL;
bit1[i] = 0LL;
}
}
void updateBit0(int r, ll v)
{
if (r <= 0)
{
return;
}
for (int i = r; i <= n; i = i + (i & (-i)))
{
bit0[i] = bit0[i] + v;
}
}
void updateBit1(int r, ll v)
{
if (r <= 0)
{
return;
}
for (int i = r; i <= n; i = i + (i & (-i)))
{
bit1[i] = bit1[i] + v;
}
}
ll queryBit0(int r)
{
ll sum = 0LL;
for (int i = r; i > 0; i = i - (i & (-i)))
{
sum = sum + bit0[i];
}
return sum;
}
ll queryBit1(int r)
{
ll sum = 0LL;
for (int i = r; i > 0; i = i - (i & (-i)))
{
sum = sum + bit1[i];
}
return sum;
}
void update(int l, int r, ll v)
{
updateBit0(l, (-1LL) * v * ((ll)(l - 1)));
updateBit0(r + 1, v * ((ll)r));
updateBit1(l, v);
updateBit1(r + 1, (-1LL) * v);
}
ll query(int l, int r)
{
ll allAmt = queryBit0(r);
ll allAdd = queryBit1(r) * ((ll)r);
ll leftAmt = queryBit0(l - 1);
ll leftAdd = queryBit1(l - 1) * ((ll)(l - 1));
return (allAmt + allAdd - leftAmt - leftAdd);
}
void solve()
{
sort(num + 1, num + (1 + n_));
ans = 0LL;
for (int i = 1; i <= n_; i++)
{
if (i > 1 && num[i - 1].first == num[i].first && (num[i - 1].second + 1) < num[i].second)
{
update(num[i - 1].second + 1, num[i].second - 1, 1LL);
}
if (maxCol[num[i].second] == num[i].first && !activeRow[num[i].second])
{
continue;
}
if (maxCol[num[i].second] == num[i].first && activeRow[num[i].second])
{
ans = ans + query(num[i].second, num[i].second);
}
if (!activeRow[num[i].second])
{
ll oldVal = query(num[i].second, num[i].second);
update(num[i].second, num[i].second, (-1LL) * oldVal);
activeRow[num[i].second] = true;
}
}
}
int main()
{
while (~scanf("%d", &n_))
{
input();
compress();
init();
solve();
ans = ans + ((ll)n_);
printf("%lld\n", ans);
}
return 0;
}