关于Dict(),Counter()踩过的一点坑

2018-06-14  本文已影响0人  不务正业的码农

首先上题目,题目思路很简单,乍看一下也是很好解决,但是在解题的过程中还是遇到不少坑,Python虽然轮子多,但是坑也不少。
这里简单反思一下解题过程中,遇到的各种各样的坑。


The bill is represented as an array with information about the calls. Help Nicola to calculate the cost for each of Sophia calls. Each call is represented as a string with date, time and duration of the call in seconds in the follow format:
"YYYY-MM-DD hh:mm:ss duration"
The date and time in this information are the start of the call.
Space-Time Communications Co. has several rules on how to calculate the cost of calls:
First 100 (one hundred) minutes in one day are priced at 1 coin per minute;
After 100 minutes in one day, each minute costs 2 coins per minute;
All calls are rounded up to the nearest minute. For example 59 sec ≈ 1 min, 61 sec ≈ 2 min;
Calls count on the day when they began. For example if a call was started 2014-01-01 23:59:59, then it counted to 2014-01-01;
For example:
2014-01-01 01:12:13 181
2014-01-02 20:11:10 600
2014-01-03 01:12:13 6009
2014-01-03 12:13:55 200
First day -- 181s≈4m -- 4 coins;
Second day -- 600s=10m -- 10 coins;
Third day -- 6009s≈101m + 200s≈4m -- 100 + 5 * 2 = 110 coins;
Total -- 124 coins.
Input: Information about calls as a tuple of strings.
Output: The total cost as an integer.
具体题目可以参见链接https://py.checkio.org/zh-hans/mission/calls-home/


最常规的解法

首先,我们获取日期信息和通话时间,再将相同日期的通话时间进行累加。最简单的想法是利用字典的键位的唯一性,
每取到一个日期的值,就将对应字典的value更新即可。但是这里有一个问题:

  1. 如果字典key存在那么,使用直接赋值和增量赋值(+=)都是没有问题的。
  2. 如果字典key不存在,那么使用直接赋值是可行的,但是使用增量赋值就会报keyerror
    此处插入一段:
    In line with the fail-fast philosophy, dict access with d[k] raises an error when k is not an existing key. Every Pythonista
    knows that d.get(k, default) is an alternative to d[k] whenever a default value is more convenient than handling KeyError.
    那么如果使字典也能实现不存在key值的时候进行增量赋值呢,这里可以使用dict.get(value,default=None)方法来实现,
    该方法返回指定键的值,如果值不在字典中返回default值。由此可以得到以下第一种解法
from math import ceil
def total_cost(calls):
  dict = {}
  for call in calls:
    date, _, length = call.split(" ")
    dict[date] = dict.get(date,0) + ceil(int(length)/60)
  return sum([i*2-100 if i>100 else i for i in dict.values()]) 

使用Built-in函数Counter()的解法

Counter()是collections下面的一个计数器函数。有兴趣的可以去看看Counter()的源码,Counter()继承dict类,因此我们可以使用dict的方法来使用counter().因为 Counter 实现了defaultdict()的__missing__ 方法, 所以当访问不存在的key的时候,返回值为0。利用这个特性,我们就可以无忧虑的使用增量赋值来达成我们的目的。

from math import ceil
from collections import Counter
def total_cost(calls):
  dict_total = Counter()
  for call in calls:
    date, _, length = call.split(" ")
    dict_total[date] += ceil(int(length)/60)
  return return sum([i*2-100 if i>100 else i for i in dict_total.values()])

当然你也可以使用counter().update方法,把所有的值放在一个可迭代对象中,然后传入update()函数进行更新,此处为了简单,就直接用了增量赋值

使用defaultdict()的解法

其实上一种方法中用counter()属于大材小用,我们需要的只是实现了__missing__方法的dict类即可,在这里我们可以调用collections模块下的defaultdict()方法来帮助我们完成。
实现原理,defaultdict()在实例化的时候,需要传入一个可调用对象,这个可调用对象在__getitem__方法找不到键的时候被调用,让__getitem__方法返回某种默认值

from collections import defaultdict
from math import ceil
def total_cost(calls):
  dict1 = defaultdict(int)
  for call in calls:
    date, _, length = call.split(" ")
    dict1[date] += ceil(int(length)/60)
  return sum([i*2-100 if i>100 else i for i in dict1.values()])

此处直接传入Int作为可调用对象是可行的,就是代码有点丑,以后学了更加pythonic的写法后再回来更新。

上一篇下一篇

猜你喜欢

热点阅读