#include <bits/stdc++.h>
using namespace std;
int ret;
int vis[112345];
vector<int> g[112345];
void dfs(int u) {
  vis[u] = 1;
  ret++;
  for (int v : g[u]) {
    if (!vis[v]) dfs(v);
  }
}
int main(void) {
  ios_base::sync_with_stdio(false);
  int n;
  cin >> n;
  for (int i = 0; i < n; i++) {
    int x;
    cin >> x;
    g[i + 1].emplace_back(x);
  }
  vector<long long> ans;
  for (int i = 1; i <= n; i++) {
    ret = 0;
    if (!vis[i]) {
      dfs(i);
      ans.emplace_back(ret);
    }
  }
  sort(ans.begin(), ans.end());
  long long last = ans.back();
  if (ans.size() > 1) {
    ans.pop_back();
    ans[ans.size() - 1] += last;
  }
  long long res = 0;
  for (long long x : ans) {
    res += x * x;
  }
  cout << res << endl;
  return 0;
}
