Đổi chỗ chữ số đầu tiên và chữ số cuối cùng của một số

Đề bài:

Viết chương trình C cho phép nhập vào một số nguyên n và thực hiện đổi vị trí của chữ cái đầu tiên và chữ cái cuối cùng.

Lưu ý trong trường hợp chữ số cuối cùng là 0 thì khi đổi chỗ sẽ được loại bỏ (ví dụ 9800 -> 809)

INPUT

1234

OUTPUT

4321

Giải:

#include<iostream>
using namespace std;
int n;
int a[100];
void doivtri() {
 int i = 0;
 while(n > 0 ) {
  a[i] = n % 10;
  n = n / 10;
  i++;
 }
 if( a[0] != 0)
  cout << a[0];
 for(int j = i-2; j >= 1; j--)
  cout << a[j];
 cout << a[i-1];
}

int main() {
 cin >> n;
 doivtri();
}