Python爬虫入门例题：抓取牛客网题目

最近做题的时候要写一些题解，在把牛客网的题目复制下来的时候，数学公式的处理比较麻烦，所以我用Python的selenium、urllib.request和BeautifulSoup4库对题目信息进行了爬取，写题解的时候时间节约了很多。

2. 前期准备

安装selenium、urllib和BeautifulSoup库。

pip3 install urllib

pip3 install selenium

pip3 install beautifulsoup4

3. 获取网页内容

以牛客网 NC204552 咪咪游戏为例。

# 导入库

import urllib.request

import bs4

import time

from bs4 import BeautifulSoup

from selenium import webdriver

# 题目属性

problemId = "204552"

# 打开浏览器，模拟登陆

# 此处用的是Chrome，如果没有安装可以替换为其他支持的浏览器

driver = webdriver.Chrome()

获取网页内容

# 获取页面内容

# 题目链接

url = f"https://ac.nowcoder.com/acm/problem/{problemId}"

# 打开网页

driver.get(url)

# 网页加载等待时间

time.sleep(3)

# 找到输入用户名和密码框，并且设置内容

username = driver.find_element_by_id('jsEmailIpt')

# 输入账号名，xxx替换为自己的账户名

username.send_keys('xxx')

time.sleep(1)

password = driver.find_element_by_id('jsPasswordIpt')

#输入密码，xxx替换为自己的密码

password.send_keys('xxx')

time.sleep(1)

# 分析网页，找到登录按钮

login = driver.find_elements_by_css_selector('div[class=col-input-login] a')[0]

# 点击按钮

login.click()

time.sleep(3)

# 格式化源代码

soup = BeautifulSoup(driver.page_source,'lxml')

# 退出浏览器

driver.quit()

存储和预处理

# 存储

data_dict = {}

# 找到主体内容

mainContent = soup.find_all(name="div", attrs={"class" :"terminal-topic"})[0]

# 去除公式的重复html元素

for each in mainContent.find_all('mrow'):

each.decompose()

for each in mainContent.find_all(name="span", attrs={"class" :"katex-html"}):

each.decompose()

# 更换换行符

for each in mainContent.find_all('br'):

each.replace_with("\n\n")

4. 内容处理

4.1. Limit

先从比较简单的信息入手，找到题目标题、时间、和内存限制。

# Limit

# 找到题目标题、时间、和内存限制

div = mainContent.find_all(name="div", attrs={"class":"subject-item-wrap"})[0].find_all("span")

# 放入字典中存储

data_dict['Title'] = f"牛客网 NC{problemId} " + soup.title.contents[0]

# Time Limit

data_dict['Time Limit'] = div[0].contents[0].split('：')[1]

# Memory Limit

data_dict['Memory Limit'] = div[1].contents[0].split('：')[1]

定义函数，处理主体内容中诡异的空格和公式的符号。

def divTextProcess(div):

"""

处理<div>标签中的文本内容

"""

# 获取文本

strBuffer = div.get_text()

# 替换公式标记

strBuffer = strBuffer.replace("{", " $").replace("}", "$ ")

# 去除多个空格

strBuffer = strBuffer.replace(" ", "")

# 去除多个换行符

strBuffer = strBuffer.replace("\n\n\n", "\n")

# 去除内容中用\xa0表示的空格

strBuffer = strBuffer.replace("\xa0", "")

# 去除首位空格

strBuffer = strBuffer.strip()

# 返回结果

return strBuffer

4.2. Problem Description

获取题目描述

# 处理题目描述

div = mainContent.find_all(name="div", attrs={"class": "subject-question"})[0]

data_dict['Problem Description'] = divTextProcess(div)

4.3. Input

输入描述

div = mainContent.find_all(name="pre")[0]

data_dict['Input'] = divTextProcess(div)

4.4. Output

输出描述

div = mainContent.find_all(name="pre")[1]

data_dict['Output'] = divTextProcess(div)

4.5. Sample Input & Onput

输入样例，用代码框环境包围。

# Input

div = mainContent.find_all(name="div", attrs={"class":"question-oi-cont"})[0]

data_dict['Sample Input'] = "```cpp" + div.get_text() + '```'

# Onput

div = mainContent.find_all(name="div", attrs={"class":"question-oi-cont"})[1]

data_dict['Sample Onput'] = "```cpp" + div.get_text() + '```'

4.6. Note

备注

# 若有备注

if len(mainContent.find_all(name="pre")) >= 5:

div = mainContent.find_all(name="pre")[-1]

data_dict['Note'] = divTextProcess(div)

4.7. Source

题目链接

data_dict['Source'] = '[' + data_dict['Title'] + ']' + '(' + url + ')'

5. 输出

for each in data_dict.keys():

print('### ' + each + '\n')

print(data_dict[each].replace("\n\n**", "**").replace("**\n\n", "**") + '\n')

下面是最后的输出结果

### Title

牛客网 NC204552 咪咪游戏

### Time Limit

C/C++ 1秒，其他语言2秒

### Memory Limit

C/C++ 524288K，其他语言1048576K

### Problem Description

牛牛最近喜欢玩咪咪游戏，于是自己写了个程序编了个游戏让牛妹来玩。游戏是这样的：

牛牛有一个长的字符串（只包26含个小写字母），他想让牛妹判断这个字符串是好的。

定义一个串是好的：这个串是由连续的mq连接而成的。

比如 $mqmq$ 说明这个串是好的， $mqmqm$ 或 $mqmqx$ 都是不好的。现在牛牛想问牛妹这个串是否是好的，如果好的输出 $Yes$ ，否则输出 $No$

### Input

第一行一个整数Q，表示询问次数

就下来Q行，一个字符串$s

### Output

Q行，每行输出 $Yes$ 或 $No$

### Sample Input

// 这里会有```cpp代码环境，在这里为了展示方便去掉了

4

mqmq

mqmqm

mqakioi

mqqmmq

### Sample Onput

Yes

No

No

No

### Note

对于 $60\%$ 的数据满足： $|s|<=10,Q<=10$ 且保证只出现m,q两个字符

对于 $100\%$ 的数据： $|s| <=10^5,Q<=10$

对于所有数据保证只出现26个英文小写字母

### Source

[牛客网 NC204552 咪咪游戏](https://ac.nowcoder.com/acm/problem/204552)