Python:火星人及组合函数combinations()

题目描述

人类终于登上了火星的土地并且见到了神秘的火星人。人类和火星人都无法理解对方的语言,但是我们的科学家发明了一种用数字交流的方法。这种交流方法是这样的,首先,火星人把一个非常大的数字告诉人类科学家,科学家破解这个数字的含义后,再把一个很小的数字加到这个大数上面,把结果告诉火星人,作为人类的回答。

火星人用一种非常简单的方式来表示数字——掰手指。火星人只有一只手,但这只手上有成千上万的手指,这些手指排成一列,分别编号为 1,2,3……。火星人的任意两根手指都能随意交换位置,他们就是通过这方法计数的。

一个火星人用一个人类的手演示了如何用手指计数。如果把五根手指——拇指、食指、中指、无名指和小指分别编号为 1,2,3,4,5,当它们按正常顺序排列时,形成了 5 位数 12345,当你交换无名指和小指的位置时,会形成 5 位数 12354,当你把五个手指的顺序完全颠倒时,会形成 54321,在所有能够形成的 120 个 5 位数中,12345 最小,它表示 1 ;12354 第二小,它表示 2 ;54321 最大,它表示 120。下表展示了只有 3 根手指时能够形成的 6 个 3 位数和它们代表的数字:

三进制数

 123
 132
 213
 231
 312
 321

代表的数字

 1
 2
 3
 4
 5
 6

现在你有幸成为了第一个和火星人交流的地球人。一个火星人会让你看他的手指,科学家会告诉你要加上去的很小的数。你的任务是,把火星人用手指表示的数与科学家告诉你的数相加,并根据相加的结果改变火星人手指的排列顺序。输入数据保证这个结果不会超出火星人手指能表示的范围。

输入描述

第一行有一个正整数 N ,表示火星人手指的数目(1≤N≤10000)。

第二行是一个正整数 M,表示要加上去的小整数(1≤M≤100)。下一行是 1 到 N 个整数的一个排列,用空格隔开,表示火星人手指的排列顺序。

输出描述

输出一行,这一行含有 N 个整数,表示改变后的火星人手指的排列顺序。每两个相邻的数中间用一个空格分开,不能有多余的空格。

输入输出样例

示例 1

输入

5
3
1 2 3 4 5

输出

1 2 4 5 3

思考: 

从当前排列开始,暴力地寻找下一个排列。对于当前排列,从后往前比较,寻找nums[i-1]<nums[i]同的位置,把nums[i-1]与i到末尾中比nums [i-1]大的最小数交换,再将i-1之后的数进行翻转(从小到大排序),可以得到比当前排列大的最小排列。

n=int(input())
m=int(input())
nums=list(map(int,input().split()))

def find_nex(nums):
  for i in range(n-1,0,-1):
    if nums[i]>nums[i-1]:   #后一位大于前一位
      for j in range(n-1,i-1,-1):
        if nums[j]>nums[i-1]: 
          nums[j],nums[i-1]=nums[i-1],nums[j]
          return nums[:i]+nums[:i-1:-1]  #nums[:i-1;-1]将i-1之后的数进行翻转
for i in range(m):      #找后面第 m 个排序
  nums=find_nex(nums)
print(' '.join([str(i) for i in nums]))

组合函数combinations()

permutations()输出的是排列,元素的排列是分先后的,“123”和“321”不同。

但是有时只需要输出组合,不用分先后,此时可以用combinations()函数。

from itertools import *
s=['1','3','2']
for element in combinations(s,2):
a=''.join(element)#把2个元素拼起来
print(a,end-'')
#输出结果为131 232

如果序列s中有相同的字符,且s是用[表示的数组,那么s中不同位置的元素被认为不同。

from itertools import *
s=[1.143.21
for element in combinations(s,2):
a=''.join(element)
print(a,end='')
#输出结果 11 13 12 13 12 32

 序列s中有相同的字符。如果要去重用集合,s用{}表示。

from itertools import *
s={'1','1','3','2'}
for element in combinations(s,2)
a=''.join(element)
print(a,end='')
#此时输出是随机的:
23 21 31
12 13 23
13 12 32

如果要去重且输出按字典序:先用set()去重,再转为list,再排序

from itertools import *
s={1,1,3,2}
t=list(set(s))
t.sort()
for element in combinations(t,2):
  a=''.join(element)
  print(a,end='')
#输出结果12 13 23

手写组合代码:暴力法


排列数需要分先后,组合数不分先后。
把求组合的代码,去掉if,然后按从小到大打印即可。
从{1,2,3、4}中选3个的组合有4种。

s=[1,2,3,4]
for i in range(4)  #循环3次,选3个数
for j in range(i+1,4): #让第2个数比第1个大
   for k in range(j+1,4):#让第3个数比第2个大
      print("%d%d%d"%(s[i],s[j],s[k]),end=",")
#输出结果123,124,134,234,

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

努力的敲码工

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值