A11442 | Bertown Subway
时间限制1s
内存限制256MB
通过 / 提交0/0
题目描述
The construction of subway in Bertown is almost finished! The President of Berland will visit this city soon to look at the new subway himself.
There are $n$ stations in the subway. It was built according to the Bertown Transport Law:
1. For each station $i$ there exists exactly one train that goes from this station. Its destination station is $p_{i}$ , possibly $p_{i}=i$ ;
2. For each station $i$ there exists exactly one station $j$ such that $p_{j}=i$ .
The President will consider the convenience of subway after visiting it. The convenience is the number of ordered pairs $(x,y)$ such that person can start at station $x$ and, after taking some subway trains (possibly zero), arrive at station $y$ ( $1<=x,y<=n$ ).
The mayor of Bertown thinks that if the subway is not convenient enough, then the President might consider installing a new mayor (and, of course, the current mayor doesn't want it to happen). Before President visits the city mayor has enough time to rebuild some paths of subway, thus changing the values of $p_{i}$ for not more than two subway stations. Of course, breaking the Bertown Transport Law is really bad, so the subway must be built according to the Law even after changes.
The mayor wants to do these changes in such a way that the convenience of the subway is maximized. Help him to calculate the maximum possible convenience he can get!
There are $n$ stations in the subway. It was built according to the Bertown Transport Law:
1. For each station $i$ there exists exactly one train that goes from this station. Its destination station is $p_{i}$ , possibly $p_{i}=i$ ;
2. For each station $i$ there exists exactly one station $j$ such that $p_{j}=i$ .
The President will consider the convenience of subway after visiting it. The convenience is the number of ordered pairs $(x,y)$ such that person can start at station $x$ and, after taking some subway trains (possibly zero), arrive at station $y$ ( $1<=x,y<=n$ ).
The mayor of Bertown thinks that if the subway is not convenient enough, then the President might consider installing a new mayor (and, of course, the current mayor doesn't want it to happen). Before President visits the city mayor has enough time to rebuild some paths of subway, thus changing the values of $p_{i}$ for not more than two subway stations. Of course, breaking the Bertown Transport Law is really bad, so the subway must be built according to the Law even after changes.
The mayor wants to do these changes in such a way that the convenience of the subway is maximized. Help him to calculate the maximum possible convenience he can get!
输入格式
The first line contains one integer number $n$ ( $1<=n<=100000$ ) — the number of stations.
The second line contains $n$ integer numbers $p_{1}$ , $p_{2}$ , ..., $p_{n}$ ( $1<=p_{i}<=n$ ) — the current structure of the subway. All these numbers are distinct.
The second line contains $n$ integer numbers $p_{1}$ , $p_{2}$ , ..., $p_{n}$ ( $1<=p_{i}<=n$ ) — the current structure of the subway. All these numbers are distinct.
输出格式
Print one number — the maximum possible value of convenience.
输入输出样例
输入 #1
3 2 1 3
输出 #1
9
输入 #2
5 1 5 4 3 2
输出 #2
17
In the first example the mayor can change $p_{2}$ to $3$ and $p_{3}$ to $1$ , so there will be $9$ pairs: $(1,1)$ , $(1,2)$ , $(1,3)$ , $(2,1)$ , $(2,2)$ , $(2,3)$ , $(3,1)$ , $(3,2)$ , $(3,3)$ .
In the second example the mayor can change $p_{2}$ to $4$ and $p_{3}$ to $5$ .
In the second example the mayor can change $p_{2}$ to $4$ and $p_{3}$ to $5$ .
C++ 编辑器
输入
输出
可保存默认模板;新题优先使用已保存模板。
当前快捷键仅展示,暂不支持修改。
- 撤销
Ctrl / ⌘ + Z - 重做
Ctrl / ⌘ + Y - 查找
Ctrl / ⌘ + F - 全选
Ctrl / ⌘ + A - 复制
Ctrl / ⌘ + C - 剪切
Ctrl / ⌘ + X - 粘贴
Ctrl / ⌘ + V - 自动排版
工具栏排版按钮 - 草稿保存
编辑时自动保存到本机
历史
提交记录
状态说明时间源码
AI
作答助手
你好,我是作答助手。可以问思路、复杂度、样例含义或代码报错原因;不会直接给出完整 AC 代码。
确定要清空代码吗?
提交通过
评测结果:Accepted