python 拼多多秒杀_关于 拼多多笔试题-简单易懂的秒杀服务

实现一个简易的秒杀服务，有3个接口：

1. 添加一个秒杀活动 addActivity(int startTime, int endTime, int goodsId, int limitQuantity)

参数说明：

* 时间区间为左闭右开：[startTime, endTime) 。数据保证 startTime < endTime，startTime 大于当前时间戳

* goodsId 一定为存在的商品id。每个goodsId至多添加一次秒杀活动。

* limitQuantity > 0

返回值有以下两种情况：

* 添加成功，返回秒杀活动ID (从0开始自增)

* 秒杀商品数量大于商品库存，添加失败，返回-1

2. 购买秒杀商品 buyGoods(int activityId, int quantity)

参数说明：

* activityId 一定是请求时存在的活动id

* quantity > 0

返回值有以下三种情况：

* 购买成功，减少库存，返回0

* 购买数量大于秒杀商品剩余数量，购买失败，返回-1

* 秒杀未开始或已结束，购买失败，返回-1

3. 获取秒杀活动列表 getActivityList() 获取当前时刻的秒杀活动列表

秒杀列表排序方式：进行中(未售罄) > 进行中(售罄) > 未开始

* 对于「进行中(未售罄)」：依次按商品人气值从高到低、商品ID从小到大排序

* 对于「进行中(已售罄)」：依次按最后卖出时间从晚到早、商品人气值从高到低、商品ID从小到大排序

* 对于「未开始」：依次按开始时间从早到晚、商品人气值从高到低、商品ID从小到大排序

* 对于已结束的秒杀，不返回。

返回秒杀活动id列表

商品拥有以下属性：

* 商品ID，32位非负整数

* 人气值，32位非负整数

* 库存，32位非负整数

现给出一串请求，每个请求的格式为：时间戳 函数名 参数。请对每个请求都输出其返回结果 (请求已经按照时间先后顺序排序过) 。

数据范围：

* 商品数量 N <= 10,000

* 请求数量 M <= 10,000

* add 数量 A <= 1,000

* buy 数量 B <= 10,000

* list 数量 L <= 100

输入描述:

第一行是两个整数 N 和 M ，分别表示商品数量和请求数量。

接下来有 N 行，每行表示一个商品，具体格式为：3个整数(被空格分隔)分别表示商品ID，人气值，库存

接下来有 M 行，每行表示一个请求，请求已经按时间戳从小到大排序。

具体格式为：时间戳 请求类型 请求参数...

* 时间戳：正整数

* 请求类型，共三种："add", "buy" 和 "list"

* 请求参数：按题目描述中的顺序，参数之间空格分隔

输出描述:

对每个请求，输出其返回值，一个请求的输出占一行：

* add：成功输出id，失败输出-1

* buy：成功输出0，失败输出-1

* list：输出活动id列表，按要求的顺序，相邻数字之间用一个空格分隔。若列表为空，则输出空行。

示例1

输入

6 13

1001 1 10

1002 1 10

1003 2 10

1004 2 10

1005 2 10

1006 3 10

1 add 2 20 1001 10

2 buy 0 1

3 buy 0 10

4 add 5 6 1002 2

5 list

6 buy 1 1

7 add 10 20 1003 11

8 add 10 20 1003 8

9 add 10 20 1004 3

10 add 11 20 1005 5

11 add 20 30 1006 1

12 buy 3 3

13 list

输出

0

0

-1

1

0 1

-1

-1

2

3

4

5

0

2 4 0 3 5

下面是我的python 解法，由于题目复杂，当时一下没A出来，后来补全了下

s = raw_input().split(" ")

n, m = int(s[0]), int(s[1])

class Good(object):

def __init__(self, good_id, pop, num):

self.good_id = good_id

self.pop = pop

self.num = num

class Activity(object):

index = 0

def __init__(self, start_time, end_time, good_object, limit):

self.start_time = start_time

self.end_time = end_time

self.good_object = good_object

self.limit = limit

self.last_sell = 0

self.activity_id = Activity.index

Activity.index += 1

goods_dict = dict()

for i in xrange(n):

s = raw_input().split(" ")

good_id = int(s[0])

pop = int(s[1])

num = int(s[2])

goods_dict[good_id] = Good(good_id, pop, num)

unstart_dict = dict()

running_dict = dict()

empty_dict = dict()

def update(moment):

del_list = []

for key, item in empty_dict.iteritems():

if item.end_time <= moment:

del_list.append(key)

for key in del_list:

del empty_dict[key]

del_list = []

for key, item in running_dict.iteritems():

if item.end_time <= moment:

del_list.append(key)

continue

if item.limit <= 0:

empty_dict[key] = item

del_list.append(key)

for key in del_list:

del running_dict[key]

del_list = []

for key, item in unstart_dict.iteritems():

if item.start_time <= moment:

running_dict[key] = item

del_list.append(key)

for key in del_list:

del unstart_dict[key]

# print "unstart_dict", str(unstart_dict)

# print "running_dict", str(running_dict)

# print "empty_dict", str(empty_dict)

for i in xrange(1, m + 1):

update(i)

s = raw_input().split(" ")

op = s[1]

if op == "add":

start_time = int(s[2])

end_time = int(s[3])

good_id = int(s[4])

limit = int(s[5])

if limit > goods_dict[good_id].num:

print -1

continue

activity = Activity(start_time, end_time, goods_dict[good_id], limit)

unstart_dict[activity.activity_id] = activity

print activity.activity_id

elif op == "buy":

activity_id = int(s[2])

quantity = int(s[3])

if activity_id in running_dict and running_dict[activity_id].limit >= quantity:

running_dict[activity_id].limit -= quantity

running_dict[activity_id].good_object.num -= quantity

running_dict[activity_id].last_sell = i

print 0

else:

print -1

elif op == "list":

def cmp_unstart(x, y):

if x.start_time == y.start_time:

if x.good_object.pop == y.good_object.pop:

return x.good_object.good_id - y.good_object.good_id

return y.good_object.pop - x.good_object.pop

return x.start_time - y.start_time

def cmp_running(x, y):

if x.good_object.pop == y.good_object.pop:

return x.good_object.good_id - y.good_object.good_id

return y.good_object.pop - x.good_object.pop

def cmp_empty(x, y):

if x.last_sell == y.last_sell:

if x.good_object.pop == y.good_object.pop:

return x.good_object.good_id - y.good_object.good_id

return y.good_object.pop - x.good_object.pop

return y.last_sell - x.last_sell

unstart_list = sorted(unstart_dict.values(), cmp=cmp_unstart)

running_list = sorted(running_dict.values(), cmp=cmp_running)

empty_list = sorted(empty_dict.values(), cmp=cmp_empty)

for activity in running_list:

print activity.activity_id,

for activity in empty_list:

print activity.activity_id,

for activity in unstart_list:

print activity.activity_id,

print ""

大家看下有没有更好的方法~