#include <vector>
#include <iostream>
using namespace std;
int main()
{
vector<vector<int> > matrix;
int n;
cout << "Please enter the size of the identity matrix" << endl;
cin >> n;
// Initialize the matrix as a n x n array of 0.
matrix = vector<vector<int> >(n, vector<int>(n,0));
// Set the diagonal to be 1s
for(unsigned int t = 0; t < n; t++)
matrix[t][t] = 1;
// Print it
for(unsigned int y = 0; y < n; y++)
{
for(unsigned int x = 0; x < n; x++)
cout << "\t" << matrix[y][x];
cout << "\n";
}
}
I2luY2x1ZGUgPHZlY3Rvcj4KI2luY2x1ZGUgPGlvc3RyZWFtPgoKdXNpbmcgbmFtZXNwYWNlIHN0ZDsKCmludCBtYWluKCkKewogICAgdmVjdG9yPHZlY3RvcjxpbnQ+ID4gbWF0cml4OwogICAgaW50IG47CgogICAgY291dCA8PCAiUGxlYXNlIGVudGVyIHRoZSBzaXplIG9mIHRoZSBpZGVudGl0eSBtYXRyaXgiIDw8IGVuZGw7CiAgICBjaW4gPj4gbjsKCiAgICAvLyBJbml0aWFsaXplIHRoZSBtYXRyaXggYXMgYSBuIHggbiBhcnJheSBvZiAwLgogICAgbWF0cml4ID0gdmVjdG9yPHZlY3RvcjxpbnQ+ID4obiwgdmVjdG9yPGludD4obiwwKSk7CgogICAgLy8gU2V0IHRoZSBkaWFnb25hbCB0byBiZSAxcwogICAgZm9yKHVuc2lnbmVkIGludCB0ID0gMDsgdCA8IG47IHQrKykKICAgICAgICBtYXRyaXhbdF1bdF0gPSAxOwogICAgICAgIAogICAgLy8gUHJpbnQgaXQKICAgIGZvcih1bnNpZ25lZCBpbnQgeSA9IDA7IHkgPCBuOyB5KyspCiAgICB7CiAgICAgICAgZm9yKHVuc2lnbmVkIGludCB4ID0gMDsgeCA8IG47IHgrKykKICAgICAgICAgICAgY291dCA8PCAiXHQiIDw8IG1hdHJpeFt5XVt4XTsKICAgICAgICBjb3V0IDw8ICJcbiI7CiAgICB9Cn0=