WU-程序员的表白 （C++代码）

#include<iostream>
using namespace std;
void print(int a)
{
	for(int i=0;i<a+1;i++)
	{
		for(int j=0;j<a+2;j++)
		{
			if(i<a)
			{
			if(j==0||j==a+1)
			{
				cout<<"*"; 
			}
			else
			cout<<" ";
		    }
		    else
		    {
		    	if(j<a+2)
		    	cout<<"*";
			}
		}
		cout<<endl;
	}
	cout<<endl;
}
int main()
{
	int a;
	while(cin>>a)
	{
		print(a);
	}
	return 0;
}