三体攻击 java 三维差分+二分+数组映射
三体人将对地球发起攻击。
为了抵御攻击,地球人派出了 A×B×C 艘战舰,在太空中排成一个 A 层 B 行 C 列的立方体。
其中,第 i 层第 j 行第 k 列的战舰(记为战舰 (i,j,k))的生命值为 d(i,j,k)。
三体人将会对地球发起 m 轮“立方体攻击”,每次攻击会对一个小立方体中的所有战舰都造成相同的伤害。
具体地,第 t 轮攻击用 7 个参数 lat,rat,lbt,rbt,lct,rct,ht 描述;
所有满足 i∈[lat,rat],j∈[lbt,rbt],k∈[lct,rct] 的战舰 (i,j,k) 会受到 ht 的伤害。
如果一个战舰累计受到的总伤害超过其防御力,那么这个战舰会爆炸。
地球指挥官希望你能告诉他,第一艘爆炸的战舰是在哪一轮攻击后爆炸的。
输入格式
第一行包括 4 个正整数 A,B,C,m;
第二行包含 A×B×C 个整数,其中第 ((i−1)×B+(j−1))×C+(k−1)+1 个数为 d(i, j, k);
第 3 到第 m+2 行中,第 (t − 2) 行包含 7 个正整数 lat, rat, lbt, rbt, lct, rct, ht。
输出格式
输出第一个爆炸的战舰是在哪一轮攻击后爆炸的。
保证一定存在这样的战舰。
数据范围
1≤A×B×C≤106,
1≤m≤106,
0≤d(i, j, k), ht≤109,
1≤lat≤rat≤A,
1≤lbt≤rbt≤B,
1≤lct≤rct≤C
层、行、列的编号都从 1 开始。
输入样例:
2 2 2 3
1 1 1 1 1 1 1 1
1 2 1 2 1 1 1
1 1 1 2 1 2 1
1 1 1 1 1 1 2
输出样例:
2
样例解释
在第 2 轮攻击后,战舰 (1,1,1) 总共受到了 2 点伤害,超出其防御力导致爆炸。
import java.util.*;
public class Main
{
static int N=1000005,A,B,C,m,x1,x2,y1,y2,z1,z2,c;
static long s[]=new long [N],b[]=new long [N*2],bp[]=new long[N*2];
static int op[][]=new int[N][7];
static int get(int i,int j,int k)
{
return (i*B+j)*C+k;
}
static boolean check(int mid)
{
b=Arrays.copyOf(bp, N*2);
for(int i=1;i<=mid;++i)
{
x1=op[i][0];x2=op[i][1];
y1=op[i][2];y2=op[i][3];
z1=op[i][4];z2=op[i][5];
c=-1*op[i][6];
b[get(x1,y1,z1)]+=c;
b[get(x1,y1,z2+1)]-=c;
b[get(x1,y2+1,z1)]-=c;
b[get(x1,y2+1,z2+1)]+=c;
b[get(x2+1,y1,z1)]-=c;
b[get(x2+1,y1,z2+1)]+=c;
b[get(x2+1,y2+1,z1)]+=c;
b[get(x2+1,y2+1,z2+1)]-=c;
}
Arrays.setAll(s,x->0);
for(int i=1;i<=A;++i)
{
for(int j=1;j<=B;++j)
{
for(int k=1;k<=C;++k)
{
int p=get(i,j,k);
s[p]=b[get(i,j,k)];
s[p]+=s[get(i,j,k-1)];
s[p]+=s[get(i,j-1,k)];
s[p]-=s[get(i,j-1,k-1)];
s[p]+=s[get(i-1,j,k)];
s[p]-=s[get(i-1,j,k-1)];
s[p]-=s[get(i-1,j-1,k)];
s[p]+=s[get(i-1,j-1,k-1)];
if(s[p]<0)return true;
}
}
}
return false;
}
public static void main(String args[])
{
Scanner sc=new Scanner(System.in);
A=sc.nextInt();
B=sc.nextInt();
C=sc.nextInt();
m=sc.nextInt();
for(int i=1;i<=A;++i)
{
for(int j=1;j<=B;++j)
{
for(int k=1;k<=C;++k)
{
int p=get(i,j,k);
s[p]=sc.nextInt();
bp[p]=s[get(i,j,k)];
bp[p]-=s[get(i,j,k-1)];
bp[p]-=s[get(i,j-1,k)];
bp[p]+=s[get(i,j-1,k-1)];
bp[p]-=s[get(i-1,j,k)];
bp[p]+=s[get(i-1,j,k-1)];
bp[p]+=s[get(i-1,j-1,k)];
bp[p]-=s[get(i-1,j-1,k-1)];
}
}
}
for(int i=1;i<=m;++i)
{
for(int j=0;j<7;++j)
{
op[i][j]=sc.nextInt();
}
}
int l=1,r=m;
while(l<r)
{
int mid=l+r>>1;
if(check(mid))r=mid;
else l=mid+1;
}
System.out.println(r);
}
}