http://star.baidu.com/forum/forum.php?mod=viewthread&tid=3186&extra=page%3D1
#include <stdio.h>
#include <string.h>
const int MAXN = 10000;
const int MAXM = 100000;
int n;
int coord[MAXN];
int heat_value[MAXN];
int is_scenic[MAXN];
int lscenic_idx[MAXN];
int rscenic_idx[MAXN];
int range_count[MAXM + 1];
//l休息区前面的下标,m休息区的下标,r休息区后面的下标,
inline int cmp_coord(int l, int m, int r)
{
if (l == -1) return 1;//前面为休息区时返回1
if (r == -1) return -1;//后面为休息区时返回-1
int d = (coord[m] - coord[l]) - (coord[r] - coord[m]);//休息区距离和前面的景区(好像也可以是休息区,两个休息区可以连着吗)的距离-休息区距离和后面的景区的距离
return d == 0 ? 0 : (d < 0 ? -1 : 1);//休息区距离和前面的景区的距离近时d < 0 返回 -1,反正为1,为0返回0
}
void update()
{
memset(range_count, 0, sizeof(range_count));
for (int s = 0; s < n; s++) {
if (is_scenic[s] == 0) {//休息区
int l = lscenic_idx[s];
int r = rscenic_idx[s];
int c = cmp_coord(l, s, r);
if (c == 0) {//距离一样
heat_value[s] = (heat_value[l] >= heat_value[r] ? heat_value[l] : heat_value[r]);
} else if (c < 0) {//休息区距离和前面的景区的距离近,或者后面不是景区时
heat_value[s] = heat_value[l];
} else {//休息区距离和后面的景区的距离近,或者前面不是景区时
heat_value[s] = heat_value[r];
}
}
range_count[heat_value[s]]++;//热度值为heat_value[s]的+1
}
}
//给lscenic_idx,rscenic_idx赋值
void init_data()
{
for (int i = 0; i < n; i++) {
lscenic_idx[i] = (is_scenic[i] == 1 ? i : (i == 0 ? -1 : lscenic_idx[i - 1]));//is_scenic[i]为休息区时lscenic_idx[i]为上一个的下标(i为0时为-1),is_scenic[i]为景点区时为下标
}
for (int i = n - 1; i >= 0; i--) {
rscenic_idx[i] = (is_scenic[i] == 1 ? i : (i == n - 1 ? -1 : rscenic_idx[i + 1]));//is_scenic[i]为休息区时lscenic_idx[i]为下一个的下标(i为最后一个下标时为-1),is_scenic[i]为景点区时为下标
}
update();
}
int query(int range)
{
int cnt = 0;
for (int i = 1; i <= range; i++) {
cnt += range_count[i];
}
return cnt;
}
int main()
{
int KK = 1;
int T, Q;
char cmd[10];
scanf("%d", &T);
while (T--) {
scanf("%d", &n);
for (int i = 0; i < n; i++) {
scanf("%d %d", &coord[i], &heat_value[i]);
is_scenic[i] = (heat_value[i] > 0 ? 1 : 0);//如果为休息区is_scenic为0
}
init_data();
printf("Case #%d:\n", KK++);
scanf("%d", &Q);
while (Q--) {
int s, chv, range;
scanf("%s", cmd);
if (cmd[0] == 'U') {
scanf("%d %d", &s, &chv);
heat_value[s] = chv;//改变风景区热度值
update();//更改休息区热度值
} else {
scanf("%d", &range);
printf("%d\n", query(range));
}
}
}
return 0;
}