蓝桥杯练习系统(算法训练)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。

相关推荐
岁忧1 小时前
(LeetCode 面试经典 150 题 ) 11. 盛最多水的容器 (贪心+双指针)
java·c++·算法·leetcode·面试·go
chao_7891 小时前
二分查找篇——搜索旋转排序数组【LeetCode】两次二分查找
开发语言·数据结构·python·算法·leetcode
秋说3 小时前
【PTA数据结构 | C语言版】一元多项式求导
c语言·数据结构·算法
Maybyy3 小时前
力扣61.旋转链表
算法·leetcode·链表
卡卡卡卡罗特5 小时前
每日mysql
数据结构·算法
chao_7896 小时前
二分查找篇——搜索旋转排序数组【LeetCode】一次二分查找
数据结构·python·算法·leetcode·二分查找
lifallen7 小时前
Paimon 原子提交实现
java·大数据·数据结构·数据库·后端·算法
lixzest7 小时前
C++ Lambda 表达式详解
服务器·开发语言·c++·算法
EndingCoder7 小时前
搜索算法在前端的实践
前端·算法·性能优化·状态模式·搜索算法
丶小鱼丶7 小时前
链表算法之【合并两个有序链表】
java·算法·链表