有一个无序整数数组,其中每个数都不相同,且每个数的取值范围为[1,N]。 设计一个时间复杂度为O(N)的算法,找出满足下面两个条件的数对的个数:
1) 相邻两个数的和也在这在这个数组之中;
2) 这两个数本身也相邻。
答: 这个数组中的数有几个特征:(1) 每个都不相同;(2) 取值范围为[1,N], 完全满足计数排序的要求。不过这里不是要排序,但是可以用相同的思想:
用一个位数组,第i位为1表示数i在数组中, 第i位为0表示数i不在数组中。这样第一个条件的验证就很简单了,只要看一个相邻两个数的和对应位是否为1即可。
下面是C++实现:
#include <bitset> #include <vector> #include <iostream> #include <set> static const int N = 100; int FindPairsCount(const std::vector<int>& vNumbers) { std::bitset<N + 1> bitArray; for(int i = 0; i < vNumbers.size(); ++i) { bitArray.set(vNumbers[i]); } int result = 0; for(int i = 1; i < vNumbers.size(); ++i) { int nSubValue = vNumbers[i] - vNumbers[i - 1]; if((1 == nSubValue || -1 == nSubValue) && bitArray.test(vNumbers[i] + vNumbers[i - 1])) { ++result; } } return result; } int main(int argc,char** argv) { int nSize = 0; while(1) { std::cout<<"请输入数组的大小,大于2且不超过100"<<std::endl; std::cin>>nSize; if(nSize >= 3 && nSize < 100) { std::cout<<"请输入"<<nSize<<"个取值范围为[1,100]的数,重复的以及不在这个范围内的都会被过滤掉:" <<std::endl; //输入数据 std::set<int> set; int iNum; while(set.size() < nSize) { std::cin>>iNum; if(iNum >= 1 && iNum <= 100) { set.insert(iNum); } } std::cout<<std::endl<<"满足条件的数对一共有:"<<FindPairsCount(std::vector<int>(set.begin(),set.end()))<<std::endl; } } return 0; }