这个不就是蚂蚁在树枝上那道题嘛。
那时候没看懂为啥,现在懂了。
借个图
第一个难点就是能够想明白:由于是第i秒每个机器人都有自己的方向,d一样,那么最终走的距离都是一样的,(i,j)与(j,i)在求距离时,都是无区别的,我要求的是每个机器人与其他所有(n-1)个机器人距离之和,再之和。我要关注的是求距离,这时候忽略碰撞,对于最终求距离无影响!!!所以就直接求解最终每个机器人最终的位置,排个序(由于涉及到求和公式推导,其中要减去前n-1的和,所以要从小到大,消除掉成为负数)。
接着就是直接求和了,其中的俩个求和公式不可以颠倒哦,你想想就知道为啥了 前n-1的和
class Solution {
public int sumDistance(int[] nums, String s, int d) {
long mod = (long)1e9 + 7;
int n = nums.length;
long ans[] = new long[n];
for(int i = 0; i < n;i++){
ans[i] = nums[i] + (s.charAt(i) == 'R' ? d : -d);
}
Arrays.sort(ans);
long res = 0,sum = 0;
for(int i = 0; i < n;i++){
res = (res + i * ans[i] - sum) % mod;
sum += ans[i];
}
return (int)res;
}
}