牛客周赛 Round 13 解题报告 | 珂学家 | 乘法原理场 + BFS上组合 + 众数贪心

矩阵转置置

https://ac.nowcoder.com/acm/contest/65507/A


前言

alt


整体评价

终于回归了周赛的5题制,还是喜欢这种。题目有难度,才会有进度。

D是道很特别的题,感觉很典,它是基于BFS基础上的乘法组合, E也是道好题,模拟贪心好像是错的,得从众数的角度去剖析。


A. 矩阵转置置

模拟即可

import java.io.*;
import java.util.*;
import java.util.stream.*;

public class Main {

    public static void main(String[] args) {
        
        Scanner sc = new Scanner(new BufferedInputStream(System.in));

        int n = sc.nextInt();
        int[][] arr = new int[n][n];
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                arr[i][j] = sc.nextInt();
            }
        }

        // 上下整体置换
        for (int i = 0, j = n - 1; i < j; i++, j--) {
            for (int k = 0; k < n; k++) {
                // swap
                int t = arr[i][k];
                arr[i][k] = arr[j][k];
                arr[j][k] = t;
            }
        }

        // 左后整体置换
        for (int i = 0, j = n - 1; i < j; i++, j--) {
            for (int k = 0; k < n; k++) {
                // swap
                int t = arr[k][i];
                arr[k][i] = arr[k][j];
                arr[k][j] = t;
            }
        }

        for (int i = 0; i < n; i++) {
            System.out.println(
                    Arrays.stream(arr[i]).mapToObj(String::valueOf)
                            .collect(Collectors.joining(" "))
            );
        }
    }

}

B. 小红买基金

设满足条件的基金m

每一个可选/不选

但至少有一个基金存在

最后为

import java.io.BufferedInputStream;
import java.util.Scanner;

public class B1 {

    static long ksm(long b, long v, long mod) {
        long r = 1l;
        while (v > 0) {
            if (v % 2 == 1) {
                r = r * b % mod;
            }
            b = b * b % mod;
            v /= 2;
        }
        return r;
    }

    public static void main(String[] args) {

        Scanner sc = new Scanner(new BufferedInputStream(System.in));

        int n = sc.nextInt(), x = sc.nextInt(), y = sc.nextInt();

        int m = 0;
        for (int i = 0; i < n; i++) {
            int u = sc.nextInt(), v = sc.nextInt();
            if (u >= x && v <= y) {
                m++;
            }
        }

        long mod = (long)1e9 + 7;
        long r = ksm(2, m, mod);

        r = ((r - 1) % mod + mod) % mod;

        System.out.println(r);
        
    }

}

C. 小红的密码修改

分类计数的题

对这几个类别进行统计,然后枚举计算

不过需要分类讨论

  • 如果某一类有且仅有一个元素, 那只能在该类中变化
  • 如果某一类超过一个元素,那可以在全范围内进行变化 注: 不能变成自己
import java.io.BufferedInputStream;
import java.util.Scanner;

public class Main {

    static int[] parse(char[] str) {
        int[] res = new int[4];
        for (char c: str) {
            if (c >= 'a' && c <= 'z') res[0]++;
            else if (c >= 'A' && c <= 'Z') res[1]++;
            else if (c >= '0' && c <= '9') res[2]++;
            else if (",.?!".indexOf(c) != -1) res[3]++;
        }
        return res;
    }

    public static void main(String[] args) {
        Scanner sc = new Scanner(new BufferedInputStream(System.in));
        int t = sc.nextInt();
        while (t-- > 0) {
            char[] str = sc.next().toCharArray();
            // 分类统计
            int[] hash = parse(str);

            long r = 0;
            for (int i = 0; i < hash.length; i++) {
                if (hash[i] == 1) {
                    if (i == 0) r += (26 - 1);
                    else if (i == 1) r += (26 - 1);
                    else if (i == 2) r += (10 - 1);
                    else if (i == 3) r += (4 - 1);
                } else {
                    r += (long)hash[i] * (26 + 26 + 10 + 4 - 1);
                }
            }
            System.out.println(r);
        }

    }

}

D. 小红的转账设置方式

这题分两部

  • 最短路计算

  • 计算总方案数

求最小总代价,这个BFS最短路就可以出来

难点在于: 总方案数

这个方案总数和边的方向有关

在保证最小代价不变的情况下,也就是保证每个点的最小路径不变(有向图)

可以观察到

  1. 图存在两种类型的边
  • 参与最短路的边
  • 没有参与最短路的边

alt

  1. 如何处理参与最短路的边

alt

参与最短路的边,实际上 属于一个集合,由集合贡献

  1. 如何处理非参与最短路的边

这类边,正反都可以,不影响结果,贡献2


分析到这里,这题的解题思路就很容易了。

  • 先BFS预处理所有的点(最短路)
  • 从点出发,找到上游的点集(边集), 贡献
  • 找到剩下的非参与最短路的点集, 贡献2
  • 乘法原理
import java.io.*;
import java.util.*;

public class Main {

    static long ksm(long b, long v, long mod) {
        long r = 1l;
        while (v > 0) {
            if (v % 2 == 1) {
                r = r * b % mod;
            }
            b = b * b % mod;
            v /= 2;
        }
        return r;
    }

    public static void main(String[] args) {
        Scanner sc = new Scanner(new BufferedInputStream(System.in));

        int n = sc.nextInt(), m = sc.nextInt();

        List<Integer>[]g = new List[n + 1];
        Arrays.setAll(g, x -> new ArrayList<>());

        int[][] edges = new int[m][2];
        for (int i = 0; i < m; i++) {
            int u = sc.nextInt(), v = sc.nextInt();
            g[u].add(v);
            g[v].add(u);

            edges[i][0] = u;
            edges[i][1] = v;
        }

        long inf = Long.MAX_VALUE / 10;
        long[] cost = new long[n + 1];
        Arrays.fill(cost, inf);

        // BFS求最短路
        Deque<Integer> deq = new ArrayDeque<>();
        deq.offer(1);
        cost[0] = cost[1] = 0;
        while (!deq.isEmpty()) {
            int cur = deq.poll();
            for (int v : g[cur]) {
                if (cost[v] > cost[cur] + 1) {
                    cost[v] = cost[cur] + 1;
                    deq.offer(v);
                }
            }
        }

        // 乘法原理阶段
        long mod = 10_0000_0007l;
        long res = 1l;
        int edgeOfShort = 0; // 参与最短路的边
        // 从2开始计算,忽略节点2
        for (int i = 2; i <= n; i++) {
            int cnt = 0;
            for (int v: g[i]) {
                // 保证上流节点
                if (cost[i] == cost[v] + 1) {
                    cnt++;
                }
            }
            edgeOfShort += cnt;

            // 2^z - 1
            long r = ksm(2l, cnt, mod);
            r = ((r - 1) % mod + mod) % mod;

            // 乘法原理
            res = res * r % mod;
        }

        int edgeOfNone = m - edgeOfShort; // 不参与最短路的边数
        res = res * ksm(2l, edgeOfNone, mod) % mod;

        long minCost = Arrays.stream(cost).sum() % mod;
        System.out.println(minCost + " " + res);

    }

}


E. 小红打boss

这题挺难贪心的,为啥这么说,因为有多种思路贪心,但是结果并非最优。

但是无论如何贪心

  • 两种不同属性依次实施
  • 尽量让大的技能伤害double

所以这个贪心,也就变成 最优配对,整体最大化的问题。

来引入一个众数思路

假设 不同属性技能的次数,分别为 a, b, c, 其中c为众数

如果 a + b <= c

分类讨论

  • 如果最大数都在集合c中,那一定可以实现a+b=n-c次
  • 如果最大数也分布于集合a,b中,那a和b直接选取c的最小元素即可, 等价于交换,也一定能实现a+b=n-c次

如果 a + b > c

由众数定理, 一定可以构建 全是不同属性的pair(若n为奇数,单独留一个尾巴),这样可以实现 n/2个最大数的加倍


结合这两个的讨论,可以得出如下的结论

mass = min(n/2, n - max(a, b, c))

import java.io.*;
import java.util.*;

public class Main {

    public static void main(String[] args) {
        Scanner sc = new Scanner(new BufferedInputStream(System.in));

        Map<String, Integer> hash = new HashMap<>();
        hash.put("ice", 0);
        hash.put("fire", 1);
        hash.put("thunder", 2);

        int n = sc.nextInt();
        int[] magic = new int[3];
        int[] attack = new int[n];

        for (int i = 0; i < n; i++) {
            String s = sc.next();
            magic[hash.get(s)]++;
            attack[i] = sc.nextInt();
        }

        Arrays.sort(attack);
        int mass = Math.max(Math.max(magic[0], magic[1]), magic[2]);
        int m = Math.min(n / 2, n - mass);
        long ans = 0;
        for (int i = 0; i < n; i++) {
            if (i >= n - m) {
                ans += attack[i];
            }
            ans += attack[i];
        }
        System.out.println(ans);

    }

}

写在最后

alt

牛客周赛解题报告系列 文章被收录于专栏

牛客周赛,定位求职笔试真题,倾向于面试算法

全部评论

相关推荐

2 2 评论
分享
牛客网
牛客企业服务