FilesExpand file tree

 master

/

solution.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

43 lines (39 loc) · 1.11 KB

 master

/

solution.py

Copy path

Top

File metadata and controls

• Code
• Blame

43 lines (39 loc) · 1.11 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

"""

TLE

"""

class LRUCache:

# @param capacity, an integer

def __init__(self, capacity):

self.capacity = capacity

self.times = {}

self.cache = {}

self.timestamp = 0

# @return an integer

def get(self, key):

self.timestamp += 1

if key in self.cache:

self.times[key] = self.timestamp

return self.cache[key]

return -1

# @param key, an integer

# @param value, an integer

# @return nothing

def set(self, key, value):

self.timestamp += 1

if key in self.cache:

self.times[key] = self.timestamp

else:

if len(self.cache) >= self.capacity:

lru_key = self.get_lru_key()

del self.cache[lru_key]

del self.times[lru_key]

self.cache[key] = value

self.times[key] = self.timestamp

def get_lru_key(self):

min_time = self.timestamp

res = None

for key in self.times:

if self.times[key] <= min_time:

res = key

min_time = self.times[key]

return res