#include <bits/stdc++.h>
#pragma GCC optimize("O3")
using namespace std;
const int N = 1e5 + 10;
vector<int> con;
int p[N];
bool vis[N];
int main() {
  ios_base::sync_with_stdio(0);
  cin.tie(0);
  int n;
  cin >> n;
  for (int i = 1; i <= n; i++) cin >> p[i];
  for (int i = 1; i <= n; i++) {
    int k = i;
    int sz = 0;
    while (!vis[k]) {
      sz++;
      vis[k] = true;
      k = p[k];
    }
    con.push_back(sz);
  }
  sort(con.begin(), con.end());
  if (con.size() >= 2) {
    con[(int)con.size() - 2] += con[(int)con.size() - 1];
    con.pop_back();
  }
  long long ans = 0;
  for (int it : con) ans += 1LL * it * it;
  cout << ans << endl;
  return 0;
}
