#include <bits/stdc++.h>
using namespace std;
int n, v[100005], nr, nod, uz[100005];
vector<int> ans;
int main() {
  ios::sync_with_stdio(false);
  cin >> n;
  for (int i = 1; i <= n; i++) cin >> v[i];
  for (int i = 1; i <= n; i++) {
    if (uz[i]) continue;
    int nod = i, nr = 0;
    while (!uz[nod]) {
      nr++;
      uz[nod] = 1;
      nod = v[nod];
    }
    if (nr) {
      ans.push_back(nr);
    }
  }
  sort(ans.begin(), ans.end());
  if (ans.size() >= 2) {
    int a = ans.back();
    ans.pop_back();
    ans.back() += a;
  }
  long long nr = 0;
  for (auto it : ans) {
    nr += 1LL * it * it;
  }
  cout << nr << '\n';
  return 0;
}
