蓝桥杯练习系统(算法训练)ALGO-947 贫穷的城市

资源限制

内存限制:256.0MB C/C++时间限制:1.0s Java时间限制:3.0s Python时间限制:5.0s

问题描述

某城市有n个小镇,编号是1~n。由于贫穷和缺乏城市规划的人才,每个小镇有且仅有一段单向的公路通往别的小镇。有一天,一辆小轿车误入了这座城市,它只能沿着公路走,它走啊走,却再也走不出这座城市了......

问如果这辆车从某个小镇出发,走了若干段公路,会到达哪个小镇。每组数据有m个询问。

输入格式

第一行两个数n、m:表示小镇数和询问数;

接下来一行n个数,第i个数Ai:表示从小镇i出发的公路会通向小镇Ai;

接下来m行,第i行有两个数Bi和Ci:询问小轿车从小镇Bi出发,走过Ci段路后会达到哪个小镇。

输出格式

一行m个数回答每个询问。

样例输入

3 3

2 1 3

1 1

2 2

3 3

样例输出

2 2 3

数据规模和约定

对于60%的数据:1<=n、m<=1000,1<=Ai、Bi<=n,0<=Ci<=1000;

对于100%的数据:1<=n、m<=100000,1<=Ai、Bi<=n,0<=Ci<=100000。

暴力,超时,仅供理解题意

cpp 复制代码
#include<iostream>
#include<vector>
using namespace std;
const int N=100005;
vector<int> v[N];
int main(){
	int n,m;
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		int ai;
		cin>>ai;
		v[i].push_back(ai);
	}
	int ans[m];
	for(int i=0;i<m;i++){
		int b,c;
		cin>>b>>c;//从b小镇出发 

		for(int j=0;j<c;j++){
			if(v[b][0]!=b){
				int next=v[b][0];
				b=next;
			}else{
				break;
			}
		} 
		
		ans[i]=b;
	} 
	for(int i=0;i<m;i++){
		cout<<ans[i]<<" ";
	}
	return 0;
} 

倍增思想

cpp 复制代码
#include<iostream>
#include<vector>
#include<math.h>
using namespace std;
const int N=100005;
int v[N][20];
int main(){
	int n,m;
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		int ai;
		cin>>ai;
		v[i][0]=ai;
	}
	//倍增思想 
	for(int i=1;i<20;i++){
		for(int j=1;j<=n;j++){
			v[j][i]=v[v[j][i-1]][i-1];
		}
	} 
	int ans[m];
	for(int i=0;i<m;i++){
		int b,c;
		cin>>b>>c;//从b小镇出发 
		int cnt=0;
		while(c){
			if(c&1){
				b=v[b][cnt];
			}
			c=c>>1;
			cnt++;
		}
		ans[i]=b;
	} 
	for(int i=0;i<m;i++){
		cout<<ans[i]<<" ";
	}
	return 0;
} 

思路:倍增思想。

在访问前算出从任意一个小镇走任意段公路最后到达的小镇,但是不是一段一段地走,而是2^0,2^1,2^2,......地走。

v[j][i]表示从j小镇开始,走2^i段公路到达的小镇。v[j][i]=v[v[j][i-1]][i-1];表示从小镇j走2^i段 =**先走2^(i-1)段到达v[v[j][i-1]]小镇,再走2^(i-1)段。**因为2^(i-1)+2^(i-1)=2^i

到后面m次访问时,将要走的段数c转换成二进制即可。例如:7=2^0+2^1+2^2,将原本要走7段优化为走3段:走2^0段,再走2^1段,最后再走2^2。

相关推荐
闪电麦坤9512 分钟前
数据结构:二维数组(2D Arrays)
数据结构·算法
凌肖战25 分钟前
力扣网C语言编程题:快慢指针来解决 “寻找重复数”
c语言·算法·leetcode
埃菲尔铁塔_CV算法1 小时前
基于 TOF 图像高频信息恢复 RGB 图像的原理、应用与实现
人工智能·深度学习·数码相机·算法·目标检测·计算机视觉
NAGNIP2 小时前
一文搞懂FlashAttention怎么提升速度的?
人工智能·算法
Codebee2 小时前
OneCode图生代码技术深度解析:从可视化设计到注解驱动实现的全链路架构
css·人工智能·算法
刘大猫262 小时前
Datax安装及基本使用
java·人工智能·算法
Gyoku Mint5 小时前
深度学习×第4卷:Pytorch实战——她第一次用张量去拟合你的轨迹
人工智能·pytorch·python·深度学习·神经网络·算法·聚类
葫三生6 小时前
如何评价《论三生原理》在科技界的地位?
人工智能·算法·机器学习·数学建模·量子计算
拓端研究室8 小时前
视频讲解:门槛效应模型Threshold Effect分析数字金融指数与消费结构数据
前端·算法