算法基础之矩阵将0所在的行列清零5

编写一个算法，若M*N矩阵中某个元素为0，则将其所在的行和列清零。

//用两个数组标记为0的行和列，遍历时如果为0就将当前的行和列设为1，当在此遍历时，如果当前的行下标或者列下标为1时直接设为0 
#include<iostream>
using namespace std; 
int arr[100][100];
int index1[100];
int index2[100];
int main()
{
	int n;
	cin>>n;
	int count=0;
	for(int i=0;i<n;i++)
	{
		for(int j=0;j<n;j++)
		{
			if(i+j!=5)
				arr[i][j]=++count;
			cout<<arr[i][j]<<" ";
			if(arr[i][j]==0)
			{
				index1[i]=1;
				index2[j]=1;
			}
		}
		cout<<endl;
	}
	for(int i=0;i<n;i++)
		for(int j=0;j<n;j++)
			if(index1[i]==1||index2[j]==1)
				arr[i][j]=0;
	for(int i=0;i<n;i++)
	{
		for(int j=0;j<n;j++)
			cout<<arr[i][j]<<" ";
		cout<<endl;
	}
	return 0;
}