[DP]1.two sum

题目描述:

Given an array of integers, return indices of the two numbers such that they add up to a specific target.

You may assume that each input would have exactly one solution, and you may not use the same element twice.

Example:

Given nums = [2, 7, 11, 15], target = 9,

Because nums[0] + nums[1] = 2 + 7 = 9,
return [0, 1].


这道题主要用的还是Python的内置函数index()

class Solution:
    def twoSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[int]
        """
        re = [0,0]
        for i in range(len(nums)):
            if (target - nums[i]) in nums[i+1:]:
                re[0] = i
                re[1] = nums[i+1:].index(target-nums[i])+(i+1)
                return re

我觉得这个方法还是挺简单的。就是效率低,不过代码短。又好理解。

这里再记录两个Python里的常用函数:
map(f,nums)
第一个参数是一个函数f,作用在num的每一个元素上。
>>> e = list(range(10))
>>> def f(num):
...     return num+2
...
>>> list(map(f,e))
[2, 3, 4, 5, 6, 7, 8, 9, 10, 11]
>>>
第二个函数是enumerate(nums),可以将Index和item一起输出
>>> list1 = [10,11,12,13,14]
>>> for index,item in enumerate(list1):
...     print(index,item)
... 
0 10
1 11
2 12
3 13
4 14
>>> 


 
Kars is tired and resentful of the narrow mindset of his village since they are content with staying where they are and are not trying to become the perfect life form. Being a top-notch inventor, Kars wishes to enhance his body and become the perfect life form. Unfortunately, n of the villagers have become suspicious of his ideas. The i -th villager has a suspicion of ai on him. Individually each villager is scared of Kars, so they form into groups to be more powerful. The power of the group of villagers from l to r be defined as f(l,r) where f(l,r)=|al−al+1|+|al+1−al+2|+…+|ar−1−ar|. Here |x−y| is the absolute value of x−y . A group with only one villager has a power of 0 . Kars wants to break the villagers into exactly k contiguous subgroups so that the sum of their power is minimized. Formally, he must find k−1 positive integers 1≤r1<r2<…<rk−1<n such that f(1,r1)+f(r1+1,r2)+…+f(rk−1+1,n) is minimised. Help Kars in finding the minimum value of f(1,r1)+f(r1+1,r2)+…+f(rk−1+1,n) . Input The first line contains a single integer t (1≤t≤100) — the number of test cases. The description of test cases follows. The first line of each test case contains two integers n,k (1≤k≤n≤100) — the number of villagers and the number of groups they must be split into. The second line of each test case contains n integers a1,a2,…,an (1≤ai≤500) — the suspicion of each of the villagers. Output For each test case, output a single integer — the minimum possible value of sum of power of all the groups i. e. the minimum possible value of f(1,r1)+f(r1+1,r2)+…+f(rk−1+1,n) . Example inputCopy 3 4 2 1 3 5 2 6 3 1 9 12 4 7 2 12 8 1 9 8 2 3 3 1 8 7 7 9 2 outputCopy 4 11 2 Note In the first test case, we will group the villagers with suspicion (1,3,5,2) into (1,3,5) and (2) . So, f(1,3)+f(4,4)=(|1−3|+|3−5|)+0=4+0=4 . In the second test case, we will group the villagers with suspicion (1,9,12,4,7,2) into (1),(9,12),(4,7,2) . So, f(1,1)+f(2,3)+f(4,6)=0+3+8=11 .
07-07
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值