输入n,m,k ,给你n个点,他们在一个一维坐标上,每个点有两个值,一个是坐标,另一个是价值,然后有m组操作,每次操作给一个坐标,意思就是把当前这个坐标的点距离他最近的k个点(相等,取id小的)的价值加起来的平均数作为当前的价值,最后输出所有新得到的权值的和<n,m<=100000 ,k <= min(10 ,n - 1)>.
思路:
直接模拟就行了,一开始wa了好多次,被自己坑了,题意明明没说给的点的坐标是有序的,但是我自己一直认为是有序的所以就各种wa啊wa啊wa,呵呵!我们先把所有的数据按照坐标排序一下,排序之后记得hash下id,然后每次操作直接模拟就行了,具体的看代码,这个题没设计到什么想法,最大的时间复杂度是100000 * 10,模拟无压力 ,不知道为什么写着感觉挺开心的,虽然题目不难。
#include<stdio.h> #include<algorithm> #define N 110000 using namespace std; typedef struct { double dis ,num; int id; }NODE; NODE node[N]; int hash[N]; bool camp(NODE a ,NODE b) { return a.dis < b.dis; } int main () { int t ,n ,m ,k ,i ,a; scanf("%d" ,&t); while(t--) { scanf("%d %d %d" ,&n ,&m ,&k); for(i = 1 ;i <= n ;i ++) { scanf("%lf %lf" ,&node[i].dis ,&node[i].num); node[i].id = i; } sort(node + 1 ,node + n + 1 ,camp); for(i = 1 ;i <= n ;i ++) hash[node[i].id] = i; double Ans = 0; while(m--) { scanf("%d" ,&a); double sum = 0; int l = hash[a] - 1 ,r = hash[a] + 1; for(i = 1 ;i <= k ;i ++) { if(l < 1) sum += node[r].num ,r ++; else if(r > n) sum += node[l].num ,l --; else if(node[hash[a]].dis - node[l].dis == node[r].dis - node[hash[a]].dis) { if(node[l].id < node[r].id) sum += node[l].num ,l--; else sum += node[r].num ,r ++; } else if(node[hash[a]].dis - node[l].dis < node[r].dis - node[hash[a]].dis) sum += node[l].num ,l--; else sum += node[r].num ,r ++; } node[hash[a]].num = sum / k; Ans += node[hash[a]].num; } printf("%.6lf " ,Ans); } return 0; }