蓝桥杯 算法训练 操作格子

一道非常直白的线段树题目,甚至不需要慵懒操作233,如果你不知道什么是线段树,可以去看我的这篇文章:线段树入门

问题描述

给定一个n个顶点,m条边的有向图(其中某些边权可能为负,但保证没有负环)。请你计算从1号点到其他点的最短路(顶点从1到n编号)。

输入格式

第一行两个整数n, m。

接下来的m行,每行有三个整数u, v, l,表示u到v有一条长度为l的边。

输出格式

共n-1行,第i行表示1号点到i+1号点的最短路。

样例输入

3 3
1 2 -1
2 3 -1
3 1 2

样例输出

-1
-2

数据规模与约定

对于10%的数据,n = 2,m = 2。

对于30%的数据,n <= 5,m <= 10。

对于100%的数据,1 <= n <= 20000,1 <= m <= 200000,-10000 <= l <= 10000,保证从任意顶点都能到达其他所有顶点。

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
#define MAXN 100005
int arr[MAXN];
struct tr{
    int l, r, mx, sum;
}Tr[MAXN << 2];
void push(int d){
    Tr[d].mx = max(Tr[d << 1].mx, Tr[d << 1 | 1].mx);
    Tr[d].sum = Tr[d << 1].sum + Tr[d << 1 | 1].sum;
}
void build(int l, int r, int d){
    Tr[d].l = l, Tr[d].r = r;
    if(l == r){
        Tr[d].mx = arr[l], Tr[d].sum = arr[l];
        return;
    }
    int mid = (l + r) >> 1, lc = d << 1, rc = d << 1 | 1;
    build(l, mid, lc);
    build(mid + 1, r, rc);
    push(d);
}
void modify(int d, int pos, int v){
    if(Tr[d].l == Tr[d].r && Tr[d].l == pos){
        Tr[d].mx = v, Tr[d].sum = v;
        return;
    }
    int mid = (Tr[d].l + Tr[d].r) >> 1, lc = d << 1, rc = d << 1 | 1;
    if(pos <= mid)modify(lc, pos, v);
    else modify(rc, pos, v);
    push(d);
}
int querymx(int l, int r, int d){
    if(Tr[d].l == l && Tr[d].r == r)return Tr[d].mx;
    int mid = (Tr[d].l + Tr[d].r) >> 1, lc = d << 1, rc = d << 1 | 1;
    if(r <= mid)return querymx(l, r, lc);  // [L, [l, r], mid, R]
    else if( l > mid )return querymx(l, r, rc);  // [L, mid, [l, r], R]
    else return max(querymx(l, mid, lc), querymx(mid + 1, r, rc));  //[L, [l, mid, r], R]
}
int querysum(int l, int r, int d){
    if(Tr[d].l == l && Tr[d].r == r)return Tr[d].sum;
    int mid = (Tr[d].l + Tr[d].r) >> 1, lc = d << 1, rc = d << 1 | 1;
    if(r <= mid)return querysum(l, r, lc);
    else if( l > mid )return querysum(l, r, rc);
    else return querysum(l, mid, lc)+ querysum(mid + 1, r, rc);
}
int main(){
    int n, m, type, p1, p2;
    scanf("%d%d", &n, &m);
    for(int i = 1; i <= n; i++)scanf("%d", &arr[i]);
    build(1, n, 1);
    for(int i = 0; i < m; i++){
        scanf("%d%d%d", &type, &p1, &p2);
        if(type == 1)modify(1, p1, p2);
        else if(type == 2)printf("%d\n", querysum(p1, p2, 1));
        else printf("%d\n", querymx(p1, p2, 1));
    }
    return 0;
}
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容