学步园

#include<iostream>
using namespace std;
void print_wave(int n)
{
    int i,j;
    for(i=1;i<=n;i++)
    {
        for(j=1;j<=i;j++)
        {
            cout<<i;
        }
        cout<<endl;
    }
    for(i=n-1;i>=1;i--)
    {
        for(j=1;j<=i;j++)
        {
            cout<<i;
        }
        cout<<endl;
    }
}
int main()
{
    int N,am,fr,i;
    cin>>N;
    while(N--)
    {
        cin>>am>>fr;
        if(N==0)
        {
            for(i=0;i<fr-1;i++)
            {
                print_wave(am);
                cout<<endl;
            }
            print_wave(am);
        }
        else
        {
            for(i=0;i<fr;i++)
            {
                print_wave(am);
                cout<<endl;
            }
        }
    }
    return 0;
}

上面的是别人的 可以AC的

下面的是我的，，，AC不了 没找到原因

#include <iostream>

using namespace std ;

int print( int n )
{
	for ( int i = 1 ; i<= n ; ++i)
	{
	
	      for ( int j = 1 ; j<= i ;++j)
	            cout << i ;
	            cout << endl ;
	}
	for ( int i=n-1; i>=1 ;--i) 
	   {
	    for ( int j = 1 ; j<=i ;--j)
	    
	    
	             cout << i ;
	             cout << endl ;
	  }

  return 0 ;
}
int main()
{
	int t ;
	cin >>  t ;
	cout << endl ;
	while(t--)
	{
  	int fr , am ;
	cin >> am >> fr;
	 if(t==0)  
        {  
            for(int i=0;i<fr-1;i++)  
            {  
                print(am);  
                cout<<endl;  
            }  
            print(am);  
        }
    else
    {
    
	while (fr--)
	{
		print(am) ;
		cout << endl;
	}
    }
   }
}