待更新
模板
求區間總和
#include <cstdio>
#include <algorithm>
using namespace std;
#define LL long long
const int maxn = 111111;
int add[maxn<<2];
LL sum[maxn<<2];
void PushUp(int k) {
// 根據子結點更新父結點
sum[k] = sum[k<<1] + sum[k<<1|1];
}
void PushDown(int k,int m) {
// 根據父結點更新子結點
if (add[k]) {
// 根據標記修改子結點
add[k<<1] += add[k];
add[k<<1|1] += add[k];
sum[k<<1] += add[k] * (m - (m >> 1));
sum[k<<1|1] += add[k] * (m >> 1);
// 取消標記
add[k] = 0;
}
}
void build(int l,int r,int k) {
// 初始化
add[k] = 0;
if (l == r) {
scanf("%lld", &sum[k]);
return ;
}
// 遞歸建樹
int m = (l + r) >> 1;
build(l, m, k << 1);
build(m+1, r, k << 1 | 1);
// 更新父結點
PushUp(k);
}
void update(int L, int R, int c, int l, int r, int k) {
if (L <= l && r <= R) {
// 如果搜索區間在目標區間之內就直接修改
add[k] += c;
sum[k] += (LL)c * (r - l + 1);
return ;
}
// 更新子結點
PushDown(k , r - l + 1);
// 遞歸更新
int m = (l + r) >> 1;
if (L <= m)
update(L, R, c, l, m, k << 1);
if (m < R)
update(L, R, c, m+1, r, k << 1 | 1);
// 更新父結點
PushUp(k);
}
LL query(int L,int R,int l,int r,int k) {
// 查詢區間總和
if (L <= l && r <= R)
// 如果搜索區間在目標區間之內就直接修改
return sum[k];
// 更新子結點
PushDown(k , r - l + 1);
// 遞歸找總和
int m = (l + r) >> 1;
LL ans = 0;
if (L <= m) ans += query(L , R , l, m, k << 1);
if (m < R) ans += query(L , R , m+1, r, k << 1 | 1);
return ans;
}
int main() {
int n, Q;
scanf("%d%d", &n, &Q);
build(1, n, 1);
while (Q--) {
char op[2];
int a , b , c;
scanf("%s",op);
if (op[0] == 'Q') {
scanf("%d%d",&a,&b);
printf("%lld\n",query(a , b , 1, n, 1));
} else {
scanf("%d%d%d",&a,&b,&c);
update(a , b , c , 1, n, 1);
}
}
return 0;
}
A - 敵兵布陣
題意
線段樹裸題。
只要做一個樹上單點的數據維護和數據查找就可以了。
代碼
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
const int maxn = 400005;
int n, ans;
struct {
int l, r, n;
}segTree[maxn];
void init() {
int k;
for (k = 1; k < n; k <<= 1);
for (int i = k; i < k << 1; ++i) {
segTree[i].l = segTree[i].r = i - k + 1;
segTree[i].n = 0;
}
for (int i = k - 1; i > 0; --i) {
segTree[i].l = segTree[2 * i].l;
segTree[i].r = segTree[2 * i + 1].r;
segTree[i].n = 0;
}
}
void insert(int i, int x, int m) {
if (x >= segTree[i].l && x <= segTree[i].r)
segTree[i].n += m;
if (segTree[i].l == segTree[i].r)
return;
int mid = (segTree[i].l + segTree[i].r) / 2;
if (x > mid)
insert(2 * i + 1, x, m);
else
insert(2 * i, x, m);
}
void find(int x, int y, int i) {
if (segTree[i].l == x && segTree[i].r == y){
ans += segTree[i].n;
return;
}
if (segTree[i].l == segTree[i].r)
return;
int mid = (segTree[i].l + segTree[i].r) / 2;
if (x > mid)
find(x, y, 2 * i + 1);
else if (y <= mid)
find(x, y, 2 * i);
else{
find(x, mid, 2 * i);
find(mid + 1, y, 2 * i + 1);
}
}
int main() {
int t, cas = 0;
scanf("%d", &t);
while (t--) {
scanf("%d", &n);
init();
int w;
for (int i = 1; i <= n; ++i) {
scanf("%d", &w);
insert(1, i, w);
}
char op[6]; int a, b;
printf("Case %d:\n", ++cas);
while (scanf("%s", op) && op[0] != 'E') {
scanf("%d%d", &a, &b);
if (op[0] == 'Q') {
ans = 0;
find(a, b, 1);
printf("%d\n", ans);
}
else if (op[0] == 'A')
insert(1, a, b);
else
insert(1, a, -b);
}
}
return 0;
}
C - A Simple Problem with Integers
題意
線段樹裸題。套模板。
代碼
#include <cstdio>
#include <algorithm>
using namespace std;
#define LL long long
const int maxn = 111111;
int add[maxn<<2];
LL sum[maxn<<2];
void PushUp(int k) {
// 根據子結點更新父結點
sum[k] = sum[k<<1] + sum[k<<1|1];
}
void PushDown(int k,int m) {
// 根據父結點更新子結點
if (add[k]) {
// 根據標記修改子結點
add[k<<1] += add[k];
add[k<<1|1] += add[k];
sum[k<<1] += add[k] * (m - (m >> 1));
sum[k<<1|1] += add[k] * (m >> 1);
// 取消標記
add[k] = 0;
}
}
void build(int l,int r,int k) {
// 初始化
add[k] = 0;
if (l == r) {
scanf("%lld", &sum[k]);
return ;
}
// 遞歸建樹
int m = (l + r) >> 1;
build(l, m, k << 1);
build(m+1, r, k << 1 | 1);
// 更新父結點
PushUp(k);
}
void update(int L, int R, int c, int l, int r, int k) {
if (L <= l && r <= R) {
// 如果搜索區間在目標區間之內就直接修改
add[k] += c;
sum[k] += (LL)c * (r - l + 1);
return ;
}
// 更新子結點
PushDown(k , r - l + 1);
// 遞歸更新
int m = (l + r) >> 1;
if (L <= m)
update(L, R, c, l, m, k << 1);
if (m < R)
update(L, R, c, m+1, r, k << 1 | 1);
// 更新父結點
PushUp(k);
}
LL query(int L,int R,int l,int r,int k) {
// 查詢區間總和
if (L <= l && r <= R)
// 如果搜索區間在目標區間之內就直接修改
return sum[k];
// 更新子結點
PushDown(k , r - l + 1);
// 遞歸找總和
int m = (l + r) >> 1;
LL ans = 0;
if (L <= m) ans += query(L , R , l, m, k << 1);
if (m < R) ans += query(L , R , m+1, r, k << 1 | 1);
return ans;
}
int main() {
int n, Q;
scanf("%d%d", &n, &Q);
build(1, n, 1);
while (Q--) {
char op[2];
int a , b , c;
scanf("%s",op);
if (op[0] == 'Q') {
scanf("%d%d",&a,&b);
printf("%lld\n",query(a , b , 1, n, 1));
} else {
scanf("%d%d%d",&a,&b,&c);
update(a , b , c , 1, n, 1);
}
}
return 0;
}
D - Mayor's posters
題意
線段樹重復染色。
只需要一個區間染色操作和一個查詢操作即可。
唯一的問題是需要簡單Hash一下。
如果完全簡單哈希的話會有問題，所以用數組緩存一下，然后判斷兩個數是否相鄰，如果不相鄰需要在他們之間插入一個任意數來分隔。
代碼
// 94ms 1112KB C++
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn = 11111;
bool hashset[maxn];
int col[maxn<<4], x[maxn<<2];
int ans, n, cnt, cur;
struct {
int l, r;
}q[maxn];
void pushdown(int rt){
if (col[rt] != -1){
col[rt<<1] = col[rt<<1|1] = col[rt];
col[rt] = -1;
}
}
void update(int L, int R, int c, int l, int r, int rt){
// 區間染色
if (L <= l && r <= R){
col[rt] = c;
return;
}
pushdown(rt);
int m = (l+r)>>1;
if (L <= m) update(L, R, c, l, m, rt<<1);
if (R > m) update(L, R, c, m+1, r, rt<<1|1);
}
void query(int l, int r, int rt){
if (col[rt] != -1){
// 根據染色更新vis（hashset）
if (!hashset[col[rt]]) ++ans;
hashset[col[rt]] = true;
return;
}
if (l == r) return;
int m = (l+r) >> 1;
query(l, m, rt<<1);
query(m+1, r, rt<<1|1);
}
void init(){
cnt = ans = 0;
scanf("%d", &n);
memset(col, -1, sizeof(col));
memset(hashset, 0, sizeof(hashset));
}
void solve(){
init();
for (int i = 0; i < n; ++i){
scanf("%d%d", &q[i].l, &q[i].r);
x[cnt++] = q[i].l, x[cnt++] = q[i].r;
}
// 給所有的區間排序，取其中不重復的
sort(x, x + cnt);
cur = 1;
for (int i = 1; i < cnt; ++i)
if (x[i] != x[i-1])
x[cur++] = x[i];
// 如果前后兩個數是不相鄰的，那就在兩個之間插入一個前一個數（其實可以是任意）
for (int i = cur-1; i >= 1; --i)
if (x[i] != x[i-1] + 1)
x[cur++] = x[i-1] + 1;
sort(x, x + cur);
for (int i = 0; i < n; ++i){
// 找每一個問題的左右邊界
int l = lower_bound(x, x + cur, q[i].l) - x;
int r = lower_bound(x, x + cur, q[i].r) - x;
// 更新區間
update(l, r, i, 0, cur, 1);
}
// 計算答案
query(0, cur, 1);
// 輸出
printf("%d\n", ans);
}
int main(){
int t;
scanf("%d", &t);
while (t--) solve();
return 0;
}
意外發現
在看LeaderBoard的時候意外發現一個大神的代碼。線段樹用DFS做，而且時間和內存都非常小，代碼也很短，26行。真是厲害的操作。
另外，對于G++和C++的提交發現，這段代碼在G++上可以跑出16ms的神級速度，而C++就會32ms。同時，快讀對于C++編譯器會起反作用，但是G++可以很快。inline對內存的優化也是這樣。
// 0ms 680KB G++
#include <cstdio>
#include <cstring>
int c, n, ans, start[10005], end[10005], vis[10005];
void dfs(int l, int r, int x){
if (!x) return;
if (r >= start[x] && l <= end[x]){
if (!vis[x]) { vis[x] = 1; ++ans; }
if (l < start[x]) dfs(l, start[x] - 1, x - 1);
if (r > end[x]) dfs(end[x] + 1, r, x - 1);
}
else dfs(l, r, x-1);
}
int main(){
scanf("%d", &c);
while (c--){
scanf("%d", &n);
for (int i = 1; i <= n; ++i)
scanf("%d%d", start+i, end+i);
memset(vis, 0, sizeof(vis));
ans = 0;
dfs(1, 10000000, n);
printf("%d\n", ans);
}
}
E - Just a Hook
題意
線段樹裸題。
區間染色求和。
第一次不是看題解而是自己套板子過的題，還行還行。
出題人估計是Dota打多了
代碼
#include <cstdio>
#include <algorithm>
#include <iostream>
using namespace std;
const int maxn = 111111;
int add[maxn<<2];
int sum[maxn<<2];
void PushUp(int k) {
sum[k] = sum[k<<1] + sum[k<<1|1];
}
void PushDown(int k,int m) {
if (add[k]) {
add[k<<1] = add[k];
add[k<<1|1] = add[k];
sum[k<<1] = add[k] * (m - (m >> 1));
sum[k<<1|1] = add[k] * (m >> 1);
add[k] = 0;
}
}
void build(int l,int r,int k) {
add[k] = 0;
if (l == r) {
sum[k] = 1;
return ;
}
int m = (l + r) >> 1;
build(l, m, k << 1);
build(m+1, r, k << 1 | 1);
PushUp(k);
}
void update(int L, int R, int c, int l, int r, int k) {
if (L <= l && r <= R) {
add[k] = c;
sum[k] = (int)c * (r - l + 1);
return ;
}
PushDown(k , r - l + 1);
int m = (l + r) >> 1;
if (L <= m)
update(L, R, c, l, m, k << 1);
if (m < R)
update(L, R, c, m+1, r, k << 1 | 1);
PushUp(k);
}
int query(int L,int R,int l,int r,int k) {
if (L <= l && r <= R)
return sum[k];
PushDown(k , r - l + 1);
int m = (l + r) >> 1;
int ans = 0;
if (L <= m) ans += query(L , R , l, m, k << 1);
if (m < R) ans += query(L , R , m+1, r, k << 1 | 1);
return ans;
}
void solve(int cas) {
int n, Q;
scanf("%d%d", &n, &Q);
build(1, n, 1);
while (Q--) {
int a, b, c;
scanf("%d%d%d", &a, &b, &c);
update(a, b, c, 1, n, 1);
}
printf("Case %d: The total value of the hook is %d.\n", cas, query(1, n, 1, n, 1));
}
int main(){
int t, cas = 0;
scanf("%d", &t);
while (t--)
solve(++cas);
return 0;
}
F - Count the Colors
題意
線段樹區間染色。
真·染色。不難但是還是去找了板子。
還是要多練習才行。
代碼
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int maxn = 8005;
int n, col[maxn<<2], vis[maxn<<2], ans[maxn<<2];
inline void pushdown(int rt){
if(col[rt] != -1){
col[rt<<1] = col[rt<<1|1] = col[rt];
col[rt] = -1;
}
}
void update(int l, int r, int data, int left, int right, int rt){
if(l <= left && right <= r){
col[rt] = data;
return;
}
if(col[rt] == data) return;
if(col[rt] != -1) pushdown(rt);
int m = (left + right) >> 1;
if(l <= m) update(l, r, data, left, m, rt<<1);
if(r > m) update(l, r, data, m+1, right, rt<<1|1);
}
void query(int left, int right, int rt){
if(col[rt] != -1){
for(int i = left; i <= right; ++i)
vis[i] = col[rt];
return;
}
if(left != right){
int m = (left + right) >> 1;
query(left, m, rt<<1);
query(m+1, right, rt<<1|1);
}
}
void init(){
int a, b, c;
memset(col, -1, sizeof(col));
memset(vis, -1, sizeof(vis));
memset(ans, 0, sizeof(ans));
//input
for (int i = 0; i < n; ++i){
scanf("%d%d%d", &a, &b, &c);
if (a >= b) continue;
update(a+1, b, c, 1, 8000, 1);
}
//calc
query(1, 8000, 1);
}
int main(){
while(~scanf("%d",&n)){
init();
int cur = 1;
while(cur < maxn){
int color = vis[cur], j = cur + 1;
if(color == -1){
++cur;
continue;
}
while(vis[j] != -1 && vis[j] == color && j < maxn)
++j;
++ans[color];
cur = j;
}
for(int i=0; i<maxn; ++i)
if(ans[i])
printf("%d %d\n",i,ans[i]);
printf("\n");
}
return 0;
}
G - Balanced Lineup
題意
線段樹區間求最值差。
我的做法是求區間最大值和區間最小值，最后作差。
方法很暴力，所以時間很長，優化空間很大。
不過是第一道純自己敲的線段樹hhh，也算是有點長進。
代碼
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn = 50005;
int col1[maxn<<2], col2[maxn<<2];
int n, m, h[maxn];
inline void pushup1(int l, int r, int k){
if (l == r || col1[k] != -1)
return;
int m = (l + r) >> 1;
pushup1(l, m, k<<1);
pushup1(m+1, r, k<<1|1);
col1[k] = max(col1[k<<1], col1[k<<1|1]);
}
inline void pushup2(int l, int r, int k){
if (l == r || col2[k] != -1)
return;
int m = (l + r) >> 1;
pushup2(l, m, k<<1);
pushup2(m+1, r, k<<1|1);
col2[k] = min(col2[k<<1], col2[k<<1|1]);
}
inline void build(int l, int r, int rt){
if (l == r){
col1[rt] = h[l];
return;
}
int m = (l+r) >> 1;
build(l, m, rt<<1);
build(m+1, r, rt<<1|1);
}
inline int query1(int b, int e, int l, int r, int rt){
if (b <= l && r <= e)
return col1[rt];
int ans = 0, m = (l + r) >> 1;
if (b <= m)
ans = max(ans, query1(b, e, l, m, rt<<1));
if (e > m)
ans = max(ans, query1(b, e, m+1, r, rt<<1|1));
return ans;
}
inline int query2(int b, int e, int l, int r, int rt){
if (b <= l && r <= e)
return col2[rt];
int ans = 0x3f3f3f3f, m = (l + r) >> 1;
if (b <= m)
ans = min(ans, query2(b, e, l, m, rt<<1));
if (e > m)
ans = min(ans, query2(b, e, m+1, r, rt<<1|1));
return ans;
}
int main(){
while (~scanf("%d%d", &n, &m)){
memset(col1, -1, sizeof(col1));
memset(col2, -1, sizeof(col2));
for (int i = 1; i <= n; ++i)
scanf("%d", h + i);
build(1, n, 1);
memcpy(col2, col1, sizeof(col1));
pushup1(1, n, 1);
pushup2(1, n, 1);
int l, r, ans, a, b;
for (int i = 1; i <= m; ++i){
scanf("%d%d", &l, &r);
a = query1(l, r, 1, n, 1), b = query2(l, r, 1, n, 1);
//cout << a << " " << b << endl;
ans = a - b;
printf("%d\n", ans);
}
}
return 0;
}
H - Can you answer these queries?
題意
線段樹區間更新。
更新是讓每個值開方。
這道題有兩個坑，一個是每個值開方最多八次，也就是說值為1的點是不需要開方的。判一下sum[k] == sum[k] == r-l+1即可。
代碼
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#define ll long long
using namespace std;
const int maxn = 100005;
ll sum[maxn<<2];
int n, m, cas;
inline void PushUp(int k) {
sum[k] = sum[k<<1] + sum[k<<1|1];
}
inline void build(int l,int r,int k) {
sum[k] = 0;
if (l == r) {
scanf("%lld", &sum[k]);
return ;
}
int m = (l + r) >> 1;
build(l, m, k << 1);
build(m+1, r, k << 1 | 1);
PushUp(k);
}
inline void update(int L, int R, int l, int r, int k) {
if (l == r) {
sum[k] = sqrt(1.0*sum[k]);
return ;
}
if (L <= l && r <= R && sum[k] == r-l+1)
return;
int m = (l + r) >> 1;
if (L <= m)
update(L, R, l, m, k << 1);
if (m < R)
update(L, R, m+1, r, k << 1 | 1);
PushUp(k);
}
inline ll query(int L,int R,int l,int r,int k) {
if (L <= l && r <= R)
return sum[k];
int m = (l + r) >> 1;
ll ans = 0;
if (L <= m)
ans += query(L , R , l, m, k << 1);
if (m < R)
ans += query(L , R , m+1, r, k << 1 | 1);
return ans;
}
void solve(){
build(1, n, 1);
scanf("%d", &m);
printf("Case #%d:\n", ++cas);
while (m--){
int op, a, b;
ll ans = 0;
scanf("%d%d%d", &op, &a, &b);
int x = min(a, b), y = max(a, b);
if (op){
ans = query(x, y, 1, n, 1);
printf("%lld\n", ans);
}
else
update(x, y, 1, n, 1);
}
printf("\n");
}
int main(){
cas = 0;
while (~scanf("%d", &n))
solve();
return 0;
}
I - Tunnel Warfare
題意
線段樹區間合并。
還是第一次見這樣的題型，之前都是染色和更新。
這個解法用了結構體，當然也可以不用。
好處是簡化了函數的傳參列表。
代碼
#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <math.h>
#include <stdlib.h>
using namespace std;
const int maxn = 50000+10;
int n,m;
int s[maxn],top;//s為模擬棧
struct node
{
int l,r;
int ls,rs,ms;//ls,左端最大連續區間，rs右端最大連續區間，ms區間內最大連續區間
} a[maxn<<2];
void init(int l,int r,int i)
{
a[i].l = l;
a[i].r = r;
a[i].ls = a[i].rs = a[i].ms = r-l+1;
if(l!=r)
{
int mid = (l+r)>>1;
init(l,mid,i*2);
init(mid+1,r,2*i+1);
}
}
void insert(int i,int t,int x)
{
if(a[i].l == a[i].r)
{
if(x==1)
a[i].ls = a[i].rs = a[i].ms = 1;//修復
else
a[i].ls = a[i].rs = a[i].ms = 0;//破壞
return ;
}
int mid = (a[i].l+a[i].r)>>1;
if(t<=mid)
insert(2*i,t,x);
else
insert(2*i+1,t,x);
a[i].ls = a[2*i].ls;//左區間
a[i].rs = a[2*i+1].rs;//右區間
a[i].ms = max(max(a[2*i].ms,a[2*i+1].ms),a[2*i].rs+a[2*i+1].ls);//父親區間內的最大區間必定是，左子樹最大區間，右子樹最大區間，左右子樹合并的中間區間，三者中最大的區間值
if(a[2*i].ls == a[2*i].r-a[2*i].l+1)//左子樹區間滿了的話，父親左區間要加上右孩子的左區間
a[i].ls += a[2*i+1].ls;
if(a[2*i+1].rs == a[2*i+1].r-a[2*i+1].l+1)//同理
a[i].rs += a[2*i].rs;
}
int query(int i,int t)
{
if(a[i].l == a[i].r || a[i].ms == 0 || a[i].ms == a[i].r-a[i].l+1)//到了葉子節點或者該訪問區間為空或者已滿都不必要往下走了
return a[i].ms;
int mid = (a[i].l+a[i].r)>>1;
if(t<=mid)
{
if(t>=a[2*i].r-a[2*i].rs+1)//因為t<=mid，看左子樹，a[2*i].r-a[2*i].rs+1代表左子樹右邊連續區間的左邊界值，如果t在左子樹的右區間內，則要看右子樹的左區間有多長并返回
return query(2*i,t)+query(2*i+1,mid+1);
else
return query(2*i,t);//如果不在左子樹的右邊界區間內，則只需要看左子樹
}
else
{
if(t<=a[2*i+1].l+a[2*i+1].ls-1)//同理
return query(2*i+1,t)+query(2*i,mid);
else
return query(2*i+1,t);
}
}
int main()
{
int i,j,x;
char ch[2];
while(~scanf("%d%d",&n,&m))
{
top = 0;
init(1,n,1);
while(m--)
{
scanf("%s",ch);
if(ch[0] == 'D')
{
scanf("%d",&x);
s[top++] = x;
insert(1,x,0);
}
else if(ch[0] == 'Q')
{
scanf("%d",&x);
printf("%d\n",query(1,x));
}
else
{
if(x>0)
{
x = s[--top];
insert(1,x,1);
}
}
}
}
return 0;
}
