C. Bertown Subway

题目解析

通过分析题目得到,车站之间回形成环。题目要求的是通过改变两个数p[i],使得车站到其他车站的可达数量最大化(便利最大化)。因为每一个i对应的pi值都是不同的,所以只能通过交换两个数来实现。

将车站到车站之间的路径可以构成环,环中的每个车站都可以到达其他的车站,若环的长度是a,则车站可达的数量为a的2次方

题目要求可达的数量最大化,则只需要让两个长度最大的环连接到一起即可。若这两个环的长度分别为a和b,则两个连接在一起形成的大环的可达数量(便利度)为(a + b)^ 2。

利用上述思路编写的代码如下:

cpp 复制代码
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
#define MAXN 100000 + 5
#define ll long long

int p[MAXN];//存储站信息
bool visited[MAXN];//用于判断当前站是否被访问了,避免重复访问环
int main()
{
    vector<ll> v;//存储所有环的长度
    int n;
    ll r = 0;//结果

    cin >> n;
    for(int i = 1; i <= n; i++)cin >> p[i];

    for(int i = 1; i <= n; i++)//i表示的是本站
    {
        if(visited[i] == true)continue;

        //利用后续的代码求环的长度,这一部分是最重要的
        ll cnt = 1;
        int x = p[i];//x是本站的目的站
        visited[x] = true;
        while(x != i)//如果说没有形成环继续执行
        {
            x = p[x];
            visited[x] = true;
            cnt++;//这个cnt是环的长度
        }

        v.push_back(cnt);//将这个环的长度放进数组中
    }
    sort(v.begin(), v.end());//从小到大进行排序
    if(v.size() >= 2)//其中有大于等于2的环数
    {
        r = v[v.size() - 2] + v[v.size() - 1];
        //现在r的值是最大的两个环长度之和
        r = r * r;//现在r的值是将两个环组合在一起的数对总和
        //(从一个站到另一个站的所有方式之和)
        for(int i = 0; i < v.size() - 2; i++)
        {
            r += v[i] * v[i];
        }
    }else//只有一个环,直接是长度的平方
    {
        r = v[0] * v[0];
    }
    cout << r;
    return 0;
}
相关推荐
CHANG_THE_WORLD5 分钟前
金字塔降低采样
算法·金字塔采样
不知天地为何吴女士2 小时前
Day32| 509. 斐波那契数、70. 爬楼梯、746. 使用最小花费爬楼梯
算法
小坏坏的大世界2 小时前
C++ STL常用容器总结(vector, deque, list, map, set)
c++·算法
励志要当大牛的小白菜5 小时前
ART配对软件使用
开发语言·c++·qt·算法
qq_513970445 小时前
力扣 hot100 Day56
算法·leetcode
PAK向日葵6 小时前
【算法导论】如何攻克一道Hard难度的LeetCode题?以「寻找两个正序数组的中位数」为例
c++·算法·面试
爱喝矿泉水的猛男8 小时前
非定长滑动窗口(持续更新)
算法·leetcode·职场和发展
YuTaoShao8 小时前
【LeetCode 热题 100】131. 分割回文串——回溯
java·算法·leetcode·深度优先
YouQian7728 小时前
Traffic Lights set的使用
算法
go546315846510 小时前
基于深度学习的食管癌右喉返神经旁淋巴结预测系统研究
图像处理·人工智能·深度学习·神经网络·算法