leetcode 1001. Grid Illumination

leetcode 1001. Grid Illumination

题意：给一个n*n的矩阵，再给一个数组，表示这个点有灯泡，并且每个点会照亮该行、列、左对角线、右对角线。再给你一个数组，表示询问。每次询问，如果这个灯泡亮着，返回1，然后关闭这个灯泡以及周围8领域的灯泡；如果没亮，返回0.

思路：重点用四个数组分别统计四个方向有没有点亮的灯泡。
 

		map<int, int>hor, ver, d1, d2;
		for (int i = 0; i < lamps.size(); i++)
		{
			hor[lamps[i][0]]++;
			ver[lamps[i][1]]++;
			d1[lamps[i][0] + lamps[i][1]]++;
			d2[lamps[i][0] - lamps[i][1]]++;
			s.insert(pair<int, int>(lamps[i][0], lamps[i][1]));

		}

能统计就简单了。

然后把点放在set里面，找起来删起来都快。

每次询问直接在set里面找，删的话也直接在set里面删。

class Solution {
public:
	vector<int> gridIllumination(int N, vector<vector<int>>& lamps, vector<vector<int>>& queries) {

		set<pair<int, int>> s;

		map<int, int>hor, ver, d1, d2;
		for (int i = 0; i < lamps.size(); i++)
		{
			hor[lamps[i][0]]++;
			ver[lamps[i][1]]++;
			d1[lamps[i][0] + lamps[i][1]]++;
			d2[lamps[i][0] - lamps[i][1]]++;
			s.insert(pair<int, int>(lamps[i][0], lamps[i][1]));

		}

		vector<int> ans;
		for (int i = 0; i < queries.size(); i++)
		{

			int x = queries[i][0];
			int y = queries[i][1];
			if (hor[x] || ver[y] || d1[x + y] || d2[x - y])
			{
				ans.push_back(1);

				for (int j = -1; j <= 1; j++)
				for (int k = -1; k <= 1; k++)
				{
					int nx = x + j;
					int ny = y + k;

					auto it = s.find(make_pair(nx, ny));

					if (it != s.end())
					{
						hor[it->first]--;
						ver[it->second]--;
						d1[it->first + it->second]--;
						d2[it->first - it->second]--;
						s.erase(*it);
					}
				}
			}
			else
				ans.push_back(0);
		}
		return ans;
	}
};