#include <bits/stdc++.h>
using namespace std;
template <class T>
bool chmax(T &a, const T &b) {
  if (a < b) {
    a = b;
    return 1;
  }
  return 0;
}
template <class T>
bool chmin(T &a, const T &b) {
  if (a > b) {
    a = b;
    return 1;
  }
  return 0;
}
template <class A, size_t N, class T>
void Fill(A (&a)[N], const T &v) {
  fill((T *)a, (T *)(a + N), v);
}
const long long int INF = 1e9 + 7;
struct UnionFind {
  std::vector<int> data;
  UnionFind(int size) : data(size, -1) {}
  bool unite(int x, int y) {
    x = root(x);
    y = root(y);
    if (x != y) {
      if (data[y] < data[x]) std::swap(x, y);
      data[x] += data[y];
      data[y] = x;
    }
    return x != y;
  }
  bool find(int x, int y) { return root(x) == root(y); }
  int root(int x) { return (data[x] < 0) ? x : data[x] = root(data[x]); }
  int size(int x) { return -data[root(x)]; }
};
int main() {
  int N;
  int p[100005];
  cin >> N;
  for (int i = 0; i < (N); i++) cin >> p[i], p[i]--;
  UnionFind uf(N + 1);
  for (int i = 0; i < (N); i++) uf.unite(i, p[i]);
  vector<long long int> v;
  for (int i = 0; i < (N); i++) {
    if (!uf.find(i, N)) {
      v.push_back(uf.size(i));
      uf.unite(i, N);
    }
  }
  sort((v).begin(), (v).end());
  if (v.size() >= 2) {
    int a = v.back();
    v.pop_back();
    int b = v.back();
    v.pop_back();
    v.push_back(a + b);
  }
  long long int ans = 0;
  for (auto a : v) ans += a * a;
  cout << ans << endl;
  return 0;
}
