整形数组找回路

最新推荐文章于 2024-06-27 08:00:00 发布

mitkook

最新推荐文章于 2024-06-27 08:00:00 发布

阅读量860

点赞数

CC 4.0 BY-SA版权

分类专栏： Interview Questions

本文链接：https://blog.youkuaiyun.com/mitkook/article/details/10248101

Interview Questions 专栏收录该内容

16 篇文章

订阅专栏

本文介绍了一种在O(N)时间内使用O(1)空间找到整数数组中循环的长度的方法。通过定义快慢两个指针，先找到循环入口，再计算循环长度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一个整形数组，存储的是下一跳的位置，就像指针一样。
比如A[0]=2 A[1]=3 A[2]=1 A[3]=1 A[4]=3
就这么跳：0 2 1 3 1 3...，于是找到了一个loop，长度为2，由1、3这两个元素组成。
题目就是给出这样的数组，返回loop长度。

要求O(N) time，O(1) space.

int findLoop(int* arr, int arrLen, int pos) {
	if (pos < 0 || pos >= arrLen) {
		return -1; // invalid pos
	}
	int ptr1 = pos; // one step ptr
	int ptr2 = pos; // two step ptr

	do {
		// validate ptr1 and ptr2 here
		ptr1 = *(arr + ptr1);
		ptr2 = *(arr + ptr2);
		ptr2 = *(arr + ptr2);
	} while (ptr1 != ptr2);

	int length = 0;
	do {
		ptr1 = *(arr + ptr1);
		length++;
	} while (ptr1 != ptr2);

	return length;
}