BestCoder9 1003 Revenge of kNN(hdu 4995) 解题报告

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=4995

题目意思：在一个一维坐标轴上，给出位置 xi 和值 vi，对于 M 次询问，每次询问给出index Qi，求出离数组下标 Qi（从1开始） 最近的 K 个点，从新计算该下标所指示的value值，即等于 K 个 最近点的value值之和 / K，如果有多个K的最近点，那么选择坐标值靠前的那组。

模拟题。首先对将位置（因为输入不一定按顺序从左至右递增排序，样例有骗人的嫌疑）从小到大排序，然后求出每一个点的 K 个最近点，最近是通过xi 离当前 xpos 的距离来判断的，不断试探。如果发现左边的点的xl 比 右边的的点 xr 离 xi的距离近，那么就选左边的点，所以要通过进一步排序比较，Judge（）函数中return point[pos].xi - point[l].xi < point[r].xi - point[pos].xi; 就是通过比较两边距离值来决定是选左边的那个点还是右边的那个点。（处理得非常地巧妙）

这个代码是学人家的，自以为看懂，写起来还是反反复复修改了很多次才成功～～～

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;

typedef __int64 LL;
const int maxn = 1e5 + 5;
double vi[maxn];
vector<int> g[maxn];
int N, M, K;

struct node
{
int id, xi;
}point[maxn];

int cmp(const node& a, const node& b)
{
return a.xi < b.xi;
}

bool Judge(int pos, int l, int r)
{
if (r > N)
return true;
if (l <= 0)
return false;
if (point[pos].xi - point[l].xi != point[r].xi - point[pos].xi)  // 离两边距离值不同，选择距离值更小的那个点
return point[pos].xi - point[l].xi < point[r].xi - point[pos].xi;
return point[l].id < point[r].id;  // 如果离两边的距离值相同，那么选择坐标值小的那个
}

void Get_KNN(int pos)   // 获得每个点的最近 k 个点是那些
{
int id = point[pos].id;
int l = pos - 1, r = pos + 1;
for (int i = 0; i < K; i++)
{
if (Judge(pos, l, r))
g[id].push_back(point[l--].id);
else
g[id].push_back(point[r++].id);
}
}

int main()
{
int T;
while (scanf("%d", &T) != EOF)
{
while (T--)
{
scanf("%d%d%d", &N, &M, &K);
for (int i = 1; i <= N; i++)
{
scanf("%d%lf", &point[i].xi, &vi[i]);
point[i].id = i;
}
sort(point+1, point+1+N, cmp);
for (int i = 1; i <= N; i++)
Get_KNN(i);
int ask;
double ans = 0;
for (int i = 0; i < M; i++)
{
scanf("%d", &ask);
double sum = 0;
for (int j = 0; j < g[ask].size(); j++)
sum += vi[g[ask][j]];
vi[ask] = sum / K;
ans += vi[ask];
}
printf("%.6lf\n", ans);
for (int j = 1; j <= N; j++)
g[j].clear();
}
}
return 0;
}

View Code