题目链接:A. The beautiful values of the palace
题意:
给一个n*n的蛇形矩阵,n为奇数。
给定m个宫殿,处在每个宫殿位置(x,y)的点有一个beautiful value等于此点蛇形矩阵上数字的各位数和。
接下来有p个询问, 给出一个矩形左下角的点(x1,y1)和右上角的点(x2,y2),问被矩形覆盖的所有宫殿的value总和是多少。
数据范围:n≤1e6; m,p≤1e5
解法:
- O(1)求出这个蛇形矩阵上(x,y)点对应的美丽值。
- 离线处理询问和宫殿的点,根据询问的点按x坐标排序后,动态插入所有x小于(或小于等于)当前询问的x坐标的宫殿的值,并对每个询问返回[y1,y2]的区间和。(树状数组维护,单点修改,区间查询和)
美丽值:先求出这个点在这个蛇形矩阵的哪一层螺旋上,模拟蛇形矩阵的赋值过程求出具体值。把结果换成各数位的数字和。
ll QV(ll x,ll y){//求美丽值
ll mi,ans=0;
x=n-x+1;y=n-y+1;
mi=min(x,min(y,min(n-x+1,n-y+1)));
if(x<=y)
ans=mi*(4*(n-1)-4*mi)+10*mi-4*n-3+x+y;
else
ans=mi*(4*n-4*mi)+2*mi+1-x-y;//模拟过程
ll res=0;
while(ans){
res+=(ans%10);ans/=10;
}
return res;
}
对于一个从(x1,y1)到(x2,y2)的矩阵,通过插入y坐标的点值,这个矩阵的结果就等于query(x2,[y1,y2])-query(x1-1,[y1,y2])。
AC代码:
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cmath>
#include <cstring>
#include <map>
#include <set>
#include <queue>
#include <string>
#include <vector>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int INF = 0x7fffffff;
const int mod = 1e9+7;
const double eps = 1e-5;
const int N = 1e6+10;
void redirect(){
#ifdef LOCAL
freopen("out.txt","r",stdin);
freopen("cmp.txt","w",stdout);
#endif
}
inline ll read(){
ll f=1,x=0;char ch;
do{ch=getchar();if(ch=='-')f=-1;}while(ch<'0'||ch>'9');
do{x=x*10+ch-'0';ch=getchar();}while(ch>='0'&&ch<='9');
return x*f;
}
struct PT{
ll x,y;ll v;
} plc[N];
struct Qrr{
int x1,y1,x2,y2;int num;ll lans,rans;
} Q[N];
bool cmp_pt(PT a,PT b){
return a.x<b.x;
}
bool cmp_qrr_1(Qrr a,Qrr b){
return a.x1<b.x1;
}
bool cmp_qrr_2(Qrr a,Qrr b){
return a.x2<b.x2;
}
bool cmp_qrr_end(Qrr a,Qrr b){
return a.num<b.num;
}
int n,m,p;ll c[N];
int lowbit(int x){
return x&(-x);
}
void update(int i,int k){
while(i<=n){
c[i]+=k;
i+=lowbit(i);
}
}
ll suf_sum(int i){
ll res=0;
while(i>0){
res+=c[i];
i-=lowbit(i);
}
return res;
}
ll Query(int l,int r){
return suf_sum(r)-suf_sum(l-1);
}
ll QV(ll x,ll y){//求美丽值
ll mi,ans=0;
x=n-x+1;y=n-y+1;
mi=min(x,min(y,min(n-x+1,n-y+1)));
if(x<=y)
ans=mi*(4*(n-1)-4*mi)+10*mi-4*n-3+x+y;
else
ans=mi*(4*n-4*mi)+2*mi+1-x-y;//模拟过程
ll res=0;
while(ans){
res+=(ans%10);ans/=10;
}
return res;
}
int main(){
redirect();
int T;scanf("%d",&T);
while(T--){
scanf("%d%d%d",&n,&m,&p);
for(int i=1;i<=m;i++){
int x,y;scanf("%lld%lld",&plc[i].x,&plc[i].y);
plc[i].v=QV(plc[i].x,plc[i].y);
}
sort(plc+1,plc+m+1,cmp_pt);
for(int i=1;i<=p;i++){
int x1,y1,x2,y2;Q[i].num=i;
scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
Q[i].x1=x1;Q[i].y1=y1;Q[i].x2=x2;Q[i].y2=y2;
}
//处理左端点
memset(c,0,sizeof(c));
sort(Q+1,Q+p+1,cmp_qrr_1);
int pt=1;
for(int i=1;i<=p;i++){
while( plc[pt].x < Q[i].x1 && pt<=m ){
update(plc[pt].y,plc[pt].v);pt++;
}
Q[i].lans=Query(Q[i].y1,Q[i].y2);
}
//处理右端点
memset(c,0,sizeof(c));
sort(Q+1,Q+p+1,cmp_qrr_2);
pt=1;
for(int i=1;i<=p;i++){
while( plc[pt].x <= Q[i].x2 && pt<=m ){
update(plc[pt].y,plc[pt].v);pt++;
}
Q[i].rans=Query(Q[i].y1,Q[i].y2);
}
sort(Q+1,Q+p+1,cmp_qrr_end);
for(int i=1;i<=p;i++){
printf("%lld\n",Q[i].rans-Q[i].lans);
}
}
return 0;
}
/*
author:dragon_bra
*/