蓝桥杯2024年真题java B组 【H.拼十字】

蓝桥杯2024年真题java B组 【H.拼十字】

原题链接:拼十字

思路:

使用树状数组或线段树解决。

先将输入的信息存入到一个n行3列的数组中,将信息排序,按照长度小到大,长相同时,宽度小到大

排序。

建立三个树状数组,维护三种颜色对应的信息,树状数组的索引就是数据的宽度,值就是有几个这样宽度的数据。

遍历数组每组数据:

当颜色为0时,假设该组数据为6 3 0,则要求的就是其他两个颜色中宽度比当前宽度大的(因为长度已经从小到大排过序),也就是去1,2树状数组中找宽度比当前3的宽度大的,就是下标大于3的(就是宽度大于三的),就是1,2树状数组中的4到无穷大(就是到N)的和。

将该组数据加到对应的树状数组0中去,就是tree0.add(arr[i][1],1)

其他两种情况同理。

该过程中的树状数组中的很多空间是无效的,但还是通过了。

code:

java 复制代码
import java.io.*;
import java.util.Arrays;
public class Main {
    static int N = 100005;
    static int MOD = 1000000007;
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StreamTokenizer in = new StreamTokenizer(br);
        PrintWriter out = new PrintWriter(new OutputStreamWriter(System.out));
        in.nextToken();
        int n = (int) in.nval;
        //数据信息,一行存储一个数据项
        int[][] arr = new int[n + 1][3];
        Tree tree0 = new Tree(N);
        Tree tree1 = new Tree(N);
        Tree tree2 = new Tree(N);
        for (int i = 1; i <= n; i++) {
            in.nextToken();
            arr[i][0] = (int) in.nval;
            in.nextToken();
            arr[i][1] = (int) in.nval;
            in.nextToken();
            arr[i][2] = (int) in.nval;
        }
        long res = 0;
        //排序,先按照长度升序排序,在按照宽度进行升序排序
        Arrays.sort(arr, (a, b) -> {
            if (a[0] != b[0]) {
                return Integer.compare(a[0], b[0]); // 按 arr[i][0] 升序
            }
            return Integer.compare(a[1], b[1]); // 如果 arr[i][0] 相同,按 arr[i][1] 升序
        });
        for (int i = 1; i <= n; i++) {
            //将当前的节点加入线段树中
            //先求和
            res %= MOD;
            if (arr[i][2] == 0){
                res += tree1.sum(arr[i][1]);
                res += tree2.sum(arr[i][1]);
                tree0.add(arr[i][1],1);
            } else if (arr[i][2] == 1) {
                res += tree0.sum(arr[i][1]);
                res += tree2.sum(arr[i][1]);
                tree1.add(arr[i][1],1);
            }else{
                res += tree0.sum(arr[i][1]);
                res += tree1.sum(arr[i][1]);
                tree2.add(arr[i][1],1);
            }
        }
        out.print(res);
        out.flush();
        out.close();
        br.close();
    }
}
class Tree{
    long[] tree;
    int N;
    public Tree(int N){
        this.N = N;
        tree = new long[N + 1];
    }
    //获取最右边的1
    public long lowBit(int i) {
        return i & -i;
    }
    public void add(int i,long  val) {
        while (i <= N) {
            tree[i] += val;
            i += lowBit(i);
        }
    }
    //计算的是原数组中的 1-i 对应的和
    public long query(int i) {
        long res = 0;
        while (i > 0) {
            res += tree[i];
            i -= lowBit(i);
        }
        return res;
    }
    public long sum(int i){
        return query(N) - query(i);
    }
}
相关推荐
GUIQU.37 分钟前
【QT】嵌入式开发:从零开始,让硬件“活”起来的魔法之旅
java·数据库·c++·qt
机器学习之心4 小时前
多目标鲸鱼优化算法(NSWOA),含46种测试函数和9个评价指标,MATLAB实现
算法·matlab·多目标鲸鱼优化算法·46种测试函数·9个评价指标
max5006005 小时前
基于Meta Llama的二语习得学习者行为预测计算模型
人工智能·算法·机器学习·分类·数据挖掘·llama
callJJ5 小时前
从 0 开始理解 Spring 的核心思想 —— IoC 和 DI(2)
java·开发语言·后端·spring·ioc·di
wangjialelele5 小时前
Linux中的线程
java·linux·jvm·c++
谷咕咕5 小时前
windows下python3,LLaMA-Factory部署以及微调大模型,ollama运行对话,开放api,java,springboot项目调用
java·windows·语言模型·llama
没有bug.的程序员5 小时前
MVCC(多版本并发控制):InnoDB 高并发的核心技术
java·大数据·数据库·mysql·mvcc
在下村刘湘6 小时前
maven pom文件中<dependencyManagement><dependencies><dependency> 三者的区别
java·maven
王哥儿聊AI6 小时前
Lynx:新一代个性化视频生成模型,单图即可生成视频,重新定义身份一致性与视觉质量
人工智能·算法·安全·机器学习·音视频·软件工程
不务专业的程序员--阿飞7 小时前
JVM无法分配内存
java·jvm·spring boot