hdu 4995 离线处理+模拟

http://acm.hdu.edu.cn/showproblem.php?pid=4995

给定一维坐标下的n个点，以及每个点的权值，有m次查询，每次将查询的x点上的权值修改为离x最近的k个点权值的平均和，有相同取序号小的。最后输出修改值的总和。

先离线处理出每个x点对应的所有最近的k个点，然后模拟即可

#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <cstring>
#include <string>
#include <queue>
#include <map>
#include <iostream>
#include <algorithm>
using namespace std;
#define RD(x) scanf("%d",&x)
#define RD2(x,y) scanf("%d%d",&x,&y)
#define RD3(x,y,z) scanf("%d%d%d",&x,&y,&z)
#define clr0(x) memset(x,0,sizeof(x))
typedef long long LL;
const int maxn = 100005;
int n,m,k;
int g[maxn][11];
double v[maxn];
struct node{
    int x,i;
}p[maxn];
bool cmp(node a,node b)
{
    return a.x < b.x;
}
bool lorr(int x,int l,int r)
{
    if(r >= n)
        return true;
    if(l < 0)
        return false;
    if(p[x].x - p[l].x != p[r].x - p[x].x)
        return p[x].x - p[l].x < p[r].x - p[x].x;
    return p[l].i < p[r].i;
}
void getK(int x)
{
    int l = x - 1,r = x + 1,id = p[x].i;
    for(int i = 0;i < k;++i)
        if(lorr(x,l,r))
            g[id][i] = p[l--].i;
        else
            g[id][i] = p[r++].i;
}
void init () {
    scanf("%d%d%d", &n, &m, &k);

    for (int i = 0; i < n; i++) {
        scanf("%d%lf", &p[i].x, &v[i]);
        p[i].i = i;
    }

    sort(p , p + n, cmp);

    for (int i = 0; i < n; i++)
        getK(i);
}
int main() {
    int _,x;RD(_);while(_--){
        init();
        double ans = 0;
        while(m--){
            RD(x);
            x--;
            double sum = 0;
            for(int i = 0;i < k;++i)
                sum += v[g[x][i]];
            v[x] = sum/k;
            ans += v[x];
        }
        printf("%.6lf\n",ans);
    }
    return 0;
}