Найти максимальный элемент и поменять его с первым элементом. через функцию пользователя

#include <iostream>
#include <ctime>
using namespace std;
 
int main()
{
    srand((int)time(0));
    int n, m, imax = 0, jmax = 0;
    cout << "n="; cin >> n;
    cout << "m="; cin >> m;
 
    double **a = new double*[n], max = 0.0;
    for (int i = 0; i < n; i++)
        a[i] = new double[m];
 
    for (int i = 0; i < n; i++)
    {
        for (int j = 0; j < m; j++)
        {
            a[i][j] = rand() % 9 + 1;
            if ((i == 0 && j == 0) || a[i][j]>max) { max = a[i][j]; imax = i; jmax = j; }
            cout << a[i][j] << " ";
        }
        cout << endl;
    }
 
    cout << "max=" << max << endl;
 
    swap(a[imax][jmax], a[n - 1][m - 1]);
 
    for (int i = 0; i < n; i++)
    {
        for (int j = 0; j < m; j++)
            cout << a[i][j] << " ";
        cout << endl;
    }
 
    for (int i = 0; i < n; i++)
        delete[] a[i];
    delete[] a;
 
    system("pause");
    return 0;
}

void func(double** a, int n, int m) {
    for (int i = 0; i < n; i++)
    {
        for (int j = 0; j < m; j++)
        {
            a[i][j] = rand() % 9 + 1;
            if ((i == 0 && j == 0) || a[i][j]>max) { max = a[i][j]; imax = i; jmax = j; }
            cout << a[i][j] << " ";
        }
        cout << endl;
    }
 
    cout << "max=" << max << endl;
 
    swap(a[imax][jmax], a[0][0]);
}

#include <bits/stdc++.h>
 
using namespace std;
 
#define int long long int
 
void func(double** a, int n, int m) {
  double max_one = a[0][0];
 
  int imax;
  int jmax;
 
  for (int i = 0; i < n; i++) {
    for (int j = 0; j < m; j++) {
      if ((i == 0 && j == 0) || a[i][j] > max_one) {
        max_one = a[i][j];
        imax = i;
        jmax = j;
      }
      cout << a[i][j] << " ";
    }
    cout << endl;
  }
 
  cout << "max=" << max_one << endl;
 
  swap(a[imax][jmax], a[0][0]);
}
 
int32_t main() {
  ios_base::sync_with_stdio(false);
  cin.tie(0);
  cout.tie(0);
  cout.precision(9);
 
  int n;
  int m;
  cin >> n >> m;
 
  double** a = new double*[n];
  for (int i = 0; i < n; ++i) {
    a[i] = new double[m];
 
    for (int j = 0; j < m; ++j)
      cin >> a[i][j];
  }
 
  func(a, n, m);
 
  for (int i = 0; i < n; ++i) {
    for (int j = 0; j < m; ++j)
      cout << a[i][j] << ' ';
 
    cout << '\n';
  }
}