Python实现数组条件判断
任务要求
对一个整数数组,如果整数数组中存在重复元素且重复元素均不相邻,函数返回字符"01";如果整数数组每个元素均不相同且偶数元素个数大于奇数元素个数,函数返回字符"02";如果整数数组均不满足上述两个条件,函数返回字符"03"。例如:数组[2, 2, 5, 3, 5],返回结果是'01'。
给定一个整数数组nums,判断其属性并返回相应的字符结果:
- 条件一:数组中存在重复元素且所有重复元素均不相邻。
- 条件二:数组中所有元素均不相同且偶数元素个数大于奇数元素个数。
- 默认条件:若上述两个条件均不满足,则返回"03"。
例如:
输入:nums = [2, 2, 5, 3, 5],输出:"01"(因为数组中存在重复元素且重复元素均不相邻)。
输入:nums = [1, 3, 5, 7],输出:"02"(因为所有元素均不相同且偶数个数为0,小于奇数个数4)。
输入:nums = [1, 2, 3, 4],输出:"03"(因为既没有重复元素,也没有偶数个数大于奇数个数)。
任务分析
任务可以分为以下几个步骤:
1.检查条件一:
- 首先遍历数组,检查是否存在重复元素。
- 如果存在重复元素,则进一步检查这些重复元素是否不相邻。
- 如果满足上述两个条件,则返回"01"。
2.检查条件二:
- 如果条件一不满足,则检查数组中所有元素是否唯一。
- 如果所有元素唯一,则统计偶数和奇数的个数。
- 如果偶数个数大于奇数个数,则返回"02"。
3.默认情况:
- 如果上述两个条件均不满足,则返回"03"。
任务实现
方法一:基础实现
def judge_array_properties(nums):
# 检查条件一:是否存在重复元素且所有重复元素均不相邻
seen = {}
has_duplicate = False
for i in range(len(nums)):
if nums[i] in seen:
# 检查重复元素是否相邻
if i - seen[nums[i]] != 1:
has_duplicate = True
else:
# 存在相邻的重复元素,直接跳过条件一
break
seen[nums[i]] = i
if has_duplicate:
return "01"
# 检查条件二:所有元素均不相同且偶数个数 > 奇数个数
if len(seen) == len(nums):
even_count = 0
odd_count = 0
for num in nums:
if num % 2 == 0:
even_count += 1
else:
odd_count += 1
if even_count > odd_count:
return "02"
# 默认情况
return "03"
# 测试案例
test_cases = [
[2, 2, 5, 3, 5], # 条件一满足,返回 "01"
[1, 3, 5, 7], # 条件二满足,返回 "02"
[1, 2, 3, 4], # 两个条件均不满足,返回 "03"
]
for case in test_cases:
print(f"数组:{case},结果: {judge_array_properties(case)}")
说明:
1.初始化变量:
- seen:用于记录每个元素及其最后一次出现的索引。
- has_duplicate:标记是否存在重复元素。
- adjacent_duplicate:标记是否存在相邻的重复元素。
2.遍历数组:
- 对于每个元素,检查其是否已经在seen中。
- 如果已经存在,则检查当前索引与之前索引的差是否为1(即是否相邻)。
- 如果相邻,则设置adjacent_duplicate标记为True并终止循环。
- 如果不相邻,则设置has_duplicate标记为True。
3.检查条件一:
- 如果has_duplicate为True且adjacent_duplicate为False,则返回"01"。
4.检查条件二:
- 如果所有元素唯一(即len(seen) == len(nums)),则统计偶数和奇数的个数。
- 如果偶数个数大于奇数个数,则返回"02"。
5.默认情况:
- 如果上述两个条件均不满足,则返回"03"。
方法二:优化实现
def judge_array_properties_optimized(nums):
# 检查条件一:是否存在重复元素且所有重复元素均不相邻
seen = {}
has_duplicate = False
adjacent_duplicate = False
for i in range(len(nums)):
if nums[i] in seen:
# 检查是否相邻
if i - seen[nums[i]] == 1:
adjacent_duplicate = True
break
else:
has_duplicate = True
seen[nums[i]] = i
if has_duplicate and not adjacent_duplicate:
return "01"
# 检查条件二:所有元素均不相同且偶数个数 > 奇数个数
if len(seen) == len(nums):
even_count = sum(1 for num in nums if num % 2 == 0)
odd_count = len(nums) - even_count
if even_count > odd_count:
return "02"
# 默认情况
return "03"
# 测试案例
test_cases = [
[2, 2, 5, 3, 5], # 条件一满足,返回 "01"
[1, 3, 5, 7], # 条件二满足,返回 "02"
[1, 2, 3, 4], # 两个条件均不满足,返回 "03"
]
for case in test_cases:
print(f"数组:{case},结果: {judge_array_properties_optimized(case)}")
运行结果:
优化点:
- 在遍历数组时,同时记录是否存在重复元素和相邻重复元素。
- 通过提前终止循环来优化性能。
运行结果
数组:[2, 2, 5, 3, 5],结果: 03
数组:[1, 3, 5, 7],结果: 03
数组:[1, 2, 3, 4],结果: 03
进程已结束,退出代码为 0