三体攻击 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);
		
	}
}
posted @ 2022-11-17 23:03  林动  阅读(45)  评论(0编辑  收藏  举报