JAVA程序设计：网格照明（LeetCode：1001）

在 N x N 的网格上，每个单元格 (x, y) 上都有一盏灯，其中 0 <= x < N 且 0 <= y < N 。

最初，一定数量的灯是亮着的。lamps[i] 告诉我们亮着的第 i 盏灯的位置。每盏灯都照亮其所在 x 轴、y 轴和两条对角线上的每个正方形（类似于国际象棋中的皇后）。

对于第 i 次查询 queries[i] = (x, y)，如果单元格 (x, y) 是被照亮的，则查询结果为 1，否则为 0 。

在每个查询 (x, y) 之后 [按照查询的顺序]，我们关闭位于单元格 (x, y) 上或其相邻 8 个方向上（与单元格 (x, y) 共享一个角或边）的任何灯。

返回答案数组 answer。每个值 answer[i] 应等于第 i 次查询 queries[i] 的结果。

 

示例：

输入：N = 5, lamps = [[0,0],[4,4]], queries = [[1,1],[1,0]]
输出：[1,0]
解释： 
在执行第一次查询之前，我们位于 [0, 0] 和 [4, 4] 灯是亮着的。
表示哪些单元格亮起的网格如下所示，其中 [0, 0] 位于左上角：
1 1 1 1 1
1 1 0 0 1
1 0 1 0 1
1 0 0 1 1
1 1 1 1 1
然后，由于单元格 [1, 1] 亮着，第一次查询返回 1。在此查询后，位于 [0，0] 处的灯将关闭，网格现在如下所示：
1 0 0 0 1
0 1 0 0 1
0 0 1 0 1
0 0 0 1 1
1 1 1 1 1
在执行第二次查询之前，我们只有 [4, 4] 处的灯亮着。现在，[1, 0] 处的查询返回 0，因为该单元格不再亮着。
 

提示：

1 <= N <= 10^9
0 <= lamps.length <= 20000
0 <= queries.length <= 20000
lamps[i].length == queries[i].length == 2

思路：没什么难点，四个map分别存储行、列、负斜率对角线和正斜率对角线上的点被多少个灯照着，再开一个set存储哪些位置上有灯，以便后面的删除。

class Solution {
	
	int[] dx= {0,-1,1,0,0,-1,-1,1,1};
	int[] dy= {0,0,0,-1,1,-1,1,-1,1};
	
    public int[] gridIllumination(int N, int[][] lamps, int[][] queries) {

    	int n=queries.length;
    	int[] ans=new int[n];
    	Set<String> st=new HashSet<>();
    	Map<Integer,Integer> map1=new HashMap<>();
    	Map<Integer,Integer> map2=new HashMap<>();
    	Map<Integer,Integer> map3=new HashMap<>();
    	Map<Integer,Integer> map4=new HashMap<>();
    	
    	for(int i=0;i<lamps.length;i++) {
    		map1.put(lamps[i][0], map1.getOrDefault(lamps[i][0], 0)+1);
    		map2.put(lamps[i][1], map2.getOrDefault(lamps[i][1], 0)+1);
    		map3.put(lamps[i][0]-lamps[i][1], map3.getOrDefault(lamps[i][0]-lamps[i][1], 0)+1);
    		map4.put(-lamps[i][0]-lamps[i][1], map4.getOrDefault(-lamps[i][0]-lamps[i][1], 0)+1);
    		String s=String.valueOf(lamps[i][0])+"#"+String.valueOf(lamps[i][1]);
    		st.add(s);
    	}
    	
    	for(int i=0;i<n;i++) {
			if(map1.getOrDefault(queries[i][0], 0)>0 || map2.getOrDefault(queries[i][1], 0)>0
					|| map3.getOrDefault(queries[i][0]-queries[i][1], 0)>0
					|| map4.getOrDefault(-queries[i][0]-queries[i][1], 0)>0)
				ans[i]=1;
			else
				ans[i]=0;
    		for(int k=0;k<9;k++) {
    			int x=queries[i][0]+dx[k];
    			int y=queries[i][1]+dy[k];
    			if(x<0 || x>=N || y<0 || y>=N)
    				continue;
    			String s=String.valueOf(x)+"#"+String.valueOf(y);
    			if(st.contains(s)) {
    				map1.put(x, map1.get(x)-1);
    				map2.put(y, map2.get(y)-1);
    				map3.put(x-y, map3.get(x-y)-1);
    				map4.put(-x-y, map4.get(-x-y)-1);
    				st.remove(s);
    			}
    		}
    	}
    	
    	return ans;
    	
    }
}