问题:
n * n的棋盘,每格上放有豆子,个数是 行数 和 列数 的和,
一次拿走 一行,或一列,
求拿走的个数。
输入:
多组测试数据。
第一行为两个整数,n,q(1 ≤ n ≤ 10^6,1 ≤ q ≤ 10^5),分别代表棋盘有n行n列,取q次
下面q行有两种形式:
R k代表取出第k行,并输出该行的豆子总数。
C k代表取出第k列,并输出该列的豆子总数。
输出:
每次取走的个数
样例输入
3 2
R 1
C 1
样例输出
9
7
1 | 2 | 3 | |
1 | 2 | 3 | 4 |
2 | 3 | 4 | 5 |
3 | 4 | 5 | 6 |
如上表,n=3,可以观察到,后一 行/列 的总个数比前一 行/列 的总个数多 n。
所以,第一行/列 总个数,s = (2+(n+1))*n/2;
第 n 行/列 的总个数 ,sn = s+(n-1)*n;
因为拿走就不存在了,所以 上述的 sn 要 减去不存在的,
假如:2、3 两 列已经拿走了,那么 任意 第 r 行都 少 了 2*r+(2+3) 个!!! (难点!!!)
因为 每个格子 都是 行 加 列。
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>
long long i[1000010],j[1000010];
int main()
{
char a;
long long n,t,x,s; // n 行 n列,t组测试数据
long long h,h1,l,l1;
while(scanf("%lld %lld",&n,&t)!=EOF)
{
h = 0,h1 = 0,l = 0,l1 = 0;
memset(i,0,sizeof(i));
memset(j,0,sizeof(j));
while(t--)
{
getchar();
scanf("%c%lld",&a,&x); // R 行 C 列
s = (3+n)*n/2; //第一行/列的个数
s += (x - 1)*n; // 原本第x 行/列的个数
if(a=='R')
{
if(i[x]==0)
{
i[x]=1;
h+=x;
h1++; // 共 取走 了几行
s-=(x*l1+l);
}
else // 如果之一行已经取走了
{
printf("0\n");
continue;
}
}
else
{
if(j[x]==0)
{
j[x]=1;
l+=x;
l1++; // 共取走了几列
s-=(x*h1+h);
}
else //如果这一列已经取走了
{
printf("0\n");
continue;
}
}
printf("%lld\n",s);
}
}
return 0;
}