主要是为了练习最简单的线段树(hh)
C++1 代码 部分超时
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int N = 1000010;
struct date
{
int l, r;
int tmax;
}dates[N * 4];
int n, k;
void build(int u, int l, int r)
{
dates[u].l = l, dates[u].r = r;
if(l == r)return;
int mid = (l + r) >> 1;
build(u << 1, l, mid);
build(u << 1 | 1, mid + 1, r);
}
void updatemax(int u)
{
dates[u].tmax = min(dates[u << 1].tmax, dates[u << 1 | 1].tmax);
}
void add(int u, int x, int y)
{
if(dates[u].l == dates[u].r && dates[u].l == x)
dates[u].tmax = y;
else {
int mid = (dates[u].l + dates[u].r) >> 1;
if(mid >= x)add(u << 1, x, y);
else add(u << 1 | 1, x, y);
updatemax(u);
}
}
int quire(int u, int l, int r)
{
int dl = dates[u].l, dr = dates[u].r;
if(l <= dl && r >= dr)return dates[u].tmax;
int tmax = 0x3f3f3f3f;
int mid = (dl + dr) >> 1;
if(l <= mid)tmax = quire(u << 1, l, r);
if(r > mid)tmax = min(tmax, quire(u << 1 | 1, l, r));
return tmax;
}
int main()
{
cin >> n;
build(1, 1, n);
for(int j = 1; j <= n; j ++)
{
int x; cin >> x;
add(1, j, x);
}
cin >> k;
for(int j = 1; j <= n; j ++)
{
int l = max(j - k, 1);
int r = min(j + k, n);
cout << quire(1, l, r) << ' ';
}
return 0;
}
c++代码2 数据输入的优化处理(没有add函数直接在update中实现)
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int N = 1000010;
struct date
{
int l, r;
int tmax;
}dates[N * 4];
int n, k;
int w[N];
void updatemax(int u)
{
dates[u].tmax = min(dates[u << 1].tmax, dates[u << 1 | 1].tmax);
}
void build(int u, int l, int r)
{
if(l == r)
{
dates[u] = {l, r, w[l]};
return;
}
else {
dates[u] = {l, r};
int mid = l + r >> 1;
build(u << 1, l, mid);
build(u << 1 | 1, mid + 1, r);
updatemax(u);
}
}
int quire(int u, int l, int r)
{
int dl = dates[u].l, dr = dates[u].r;
if(l <= dl && r >= dr)return dates[u].tmax;
int tmax = 0x3f3f3f3f;
int mid = (dl + dr) >> 1;
if(l <= mid)tmax = quire(u << 1, l, r);
if(r > mid)tmax = min(tmax, quire(u << 1 | 1, l, r));
return tmax;
}
int main()
{
cin >> n;
for(int j = 1; j <= n; j ++)
cin >> w[j];
build(1, 1, n);
cin >> k;
for(int j = 1; j <= n; j ++)
{
int l = max(j - k, 1);
int r = min(j + k, n);
cout << quire(1, l, r) << ' ';
}
return 0;
}