蚂蚁笔试-工程研发 2022.9.15

image.png

选O(max(n,m))


image.png

选第二个,shell附带选项和参数两者shell命令都可以带


image.png

不会
image.png
image.png

image.png

image.png

image.png

image.png

编程题1:小红的字符生成

题目描述
小红每次可以把一个字符编程两个字母表中比它小一位的字符,例如可以把'b'变成两个'a',可以把'z'变成两个'y'。
小红希望最终可以生成x个'a',你能帮小红求出初始的字符串吗?请你输出最短的合法字符串,有多解时输出任意即可。
输入描述

一个正整数x,代表最终'a'的数量
1<=x<=1000

输出描述

一个字符串,代表小红操作前的字符串。如果有多个合法解,输出任意一个合法字符串即可。但需要保证输出的是最短字符串。

示例1
输入

5

输出

ca

说明

'ca'->'bba'->'aaaaa'
输出ac也是可以的

x = int(input())

word = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'g', 'k', 'l', 'm', 'n',
       'o', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z']
word_cont = []
word_num = len(word)

num = 1
for _ in range(len(word)):
    word_cont.append(num)
    num *= 2

# print(word_cont)
res = []
while x>0:
    idx = None
    for i in range(word_num-1, -1, -1):
        if x>=word_cont[i]:
            idx = i
            break
    
    res.append(word[idx])
    x -= word_cont[idx]

print(''.join(res))

编程题2:树上节点

题目描述
给定一课根为1号节点的树,每个节点初始的权值为1.现在每次可以选择一个节点,使得其子树所有节点的权值加1,请问最少多少次操作可以使得每个节点的权值等于它的编号?
输入描述

第一行输入一个正整数n,代表树上的节点数量。
接下来的n-1行,每行输入两个正整数u和v,代表u号节点和v号节点有一条边相连。
2<=x<=100000
1<=u,v<=n

输出描述

一个整数,代表最小的操作次数

示例1
输入

3
1 2
1 3

输出

3

说明

第1次选择编号为2的子树。
第2次选择编号为3的子树。
第3次选择编号为3的子树。

说明:有两个坑,一是需要考虑输入的边是双向的,使用一个数组去标记是否访问过;二是需要考虑父节点的编号大于子节点,这样永远无法实现题目的情况,没有答案,输出-1。

# 第一种写法,自己建树,就是写起来比较麻烦
class TreeNode:
    def __init__(self, order, num=1):
        self.order = order
        self.num = num
        self.childs = []
        
n = int(input())
d = {}

for _ in range(n-1):
    u, v = map(lambda x: int(x), input().split())
    if u in d:
        u_node = d[u]
    else:
        u_node = TreeNode(u, 1)
    
    if v in d:
        v_node = d[v]
    else:
        v_node = TreeNode(v, 1)
        
    u_node.childs.append(v_node)
    v_node.childs.append(u_node)
    d[u] = u_node
    d[v] = v_node


vis = [True]+[False]*n

def dfs(node: TreeNode, num: int):
    global count

    vis[node.order] = True
    cnt = 0

    add_num = 0
    add_num += node.order - num - node.num
    cnt += add_num

    if add_num<0:
        return -1

    if len(node.childs)==0:
        return 0

    for ch in node.childs:
        if not vis[ch.order]:
            ch_cnt = dfs(ch, num+add_num)
            if ch_cnt<0:
                return -1
            cnt += ch_cnt

    return cnt

root = d[1]
print(dfs(root, 0))

# 简化版,使用列表模拟树
n = int(input())

nodes = [[] for _ in range(n+1)]
for _ in range(n-1):
    u, v = map(lambda x: int(x), input().split())

    nodes[u].append(v)
    nodes[v].append(u)


vis = [True]+[False]*n

def dfs(order, num: int):
    vis[order] = True
    cnt = 0

    add_num = 0
    add_num += order-num-1

    cnt += add_num

    if add_num<0:
        return -1

    for ch_order in nodes[order]:
        if not vis[ch_order]:
            ch_cnt = dfs(ch_order, num+add_num)
            if ch_cnt<0:
                return -1
            cnt += ch_cnt

    return cnt

print(dfs(1, 0))

编程题3:小红的好串

题目描述
小红定义一个字符串是好串,当且仅当只有一个字符串出现的次数为奇数,其它字母均为偶数。
小红拿到了一个字符串,她想知道该字符串有多少子串是好串?
子串的定义:一个字符串取一段连续的区间得到的字符串。例如“arcaea”的子串,有“arc”、“ca”等,但“ara”则不是它的子串。
输入描述

一个长度不超过200000的、仅由小写字母组成的字符串。

输出描述

好子串的数量。

示例1
输入

ababa

输出

9

说明

所有长度为1、3、5个子串均为好子串。共有9个好子串。

leetcode简单变形题:1915 https://leetcode.cn/problems/number-of-wonderful-substrings/

# 不会:前缀异或和

from collections import defaultdict

data = input()

freq = defaultdict(int)
freq[0] = 1

ans = 0
mask = 0

for ch in data: # 遍历所有的字符
    idx = ord(ch)-ord('a') # 计算字符索引
    mask ^= (1<<idx) # 计算到当前字符所有字符的掩码

    for i in range(26): # 总共26位
        mask_pre = mask^(1<<i) # 计算某一位出现次数为奇数,其它为出现次数为偶数的数量
        ans += freq[mask_pre] # 获取变位后数量

    freq[mask] += 1

print(ans)
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 216,651评论 6 501
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,468评论 3 392
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 162,931评论 0 353
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,218评论 1 292
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,234评论 6 388
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,198评论 1 299
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,084评论 3 418
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,926评论 0 274
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,341评论 1 311
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,563评论 2 333
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,731评论 1 348
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,430评论 5 343
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,036评论 3 326
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,676评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,829评论 1 269
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,743评论 2 368
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,629评论 2 354

推荐阅读更多精彩内容