C# 计算两两坐标之间的距离(SIMD加速)

问题描述:对于Point[] points坐标数组(Point是坐标点,包含X与Y),请计算两两坐标点之间的距离。

思路:既然是算距离,没有别的要求,只需要最简单的曼哈顿距离就行。

测试数据:

cs 复制代码
Random random = new Random();
int n = 3000;
Point[] points = new Point[n];
for (int i = 0; i < n; i++)
{
    points[i] = new Point(random.Next(10), random.Next(10));
}

方法一:采用普通遍历法

cs 复制代码
Stopwatch sw = Stopwatch.StartNew();
int m = (n * n - n) / 2;
List<int> list2 = new List<int>(m);
for (int i = 0; i < n; i++)
{
    Point p= points[i];
    for (int j = i + 1; j < n; j++)
    {
        Point p2 = points[j];
        int dif = Math.Abs(p.X - p2.X) + Math.Abs(p.Y - p2.Y);
        list2.Add(dif);
    }
}
sw.Stop();
Console.WriteLine($"普通耗时:{sw.ElapsedMilliseconds}ms");

方法二:采用SIMD加速

cs 复制代码
 public static int[] CalculateDistance(Point[] points)
        {
            int n = points.Length;
            int m = (n * n - n) >> 1;
            int[] ints = new int[m];
            int k = 0;
            if (n <= 4)// 如果数据量小于等于4采用指针处理
            {
                fixed (Point* ptr = points)
                {
                    for (int i = 0; i < n; i++)
                    {
                        Point* p1 = ptr + i;
                        for (int j = i + 1; j < n; j++)
                        {
                            Point* p2 = ptr + j;
                            int dif = Math.Abs(p1->X - p2->X) + Math.Abs(p1->Y - p2->Y);
                            ints[k++] = dif;
                        }
                    }
                    return ints;
                }
            }
            int vsize = Vector128<int>.Count;
            int t = 0;
            int[] xs = new int[n];
            int[] ys = new int[n];
            fixed (Point* ptr = points)
            {
                fixed (int* xptr = xs, yptr = ys, rptr = ints)
                {
                    // 分离x与y
                    for (int i = 0; i < n; i++)
                    {
                        Point* p = ptr + i;
                        *(xptr + i) = p->X;
                        *(yptr + i) += p->Y;
                    }
                    // 开始向量计算
                    for (int i = 0; i < n; i += vsize)
                    {
                        Vector128<int> vx1 = *(Vector128<int>*)(xptr + i);
                        Vector128<int> vy1 = *(Vector128<int>*)(yptr + i);
                        for (int j = i + 1; j < n; j++)
                        {
                            Vector128<int> vx2 = *(Vector128<int>*)(xptr + j);
                            Vector128<int> vy2 = *(Vector128<int>*)(yptr + j);
                            // 向量运算
                            var vDifference = Sse2.Add(Vector128.Abs(Sse2.Subtract(vx1, vx2)), Vector128.Abs(Sse2.Subtract(vy1, vy2)));
                            int* maskPtr = (int*)&vDifference;
                            int h = n - j;
                            for (int a = 0; a < vsize; a++)
                            {
                                if (a >= h) break;
                                *(rptr + t++) = *(maskPtr + a);
                            }
                        }
                    }
                }
            }
            return ints;
        }

注:根据CPU选择合适的Vector,比如Vector256、Vector512等等。本文只用Vector128.

测试结果(Release环境):

|----------|------|------|------|-------|-------|
| 数据量 | 3000 | 5000 | 7000 | 10000 | 30000 |
| 普通耗时(ms) | 43 | 127 | 211 | 538 | 3872 |
| 向量耗时(ms) | 19 | 56 | 68 | 190 | 1092 |

结论:明显向量运算性能优于普通计算。

相关推荐
Y.O.U..6 分钟前
今日八股——C++
开发语言·c++·面试
weixin_3077791315 分钟前
使用C#实现从Hive的CREATE TABLE语句中提取分区字段名和数据类型
开发语言·数据仓库·hive·c#
Xiaok101823 分钟前
解决 Hugging Face SentenceTransformer 下载失败的完整指南:ProxyError、SSLError与手动下载方案
开发语言·神经网络·php
绿草在线25 分钟前
Mock.js虚拟接口
开发语言·javascript·ecmascript
go_bai36 分钟前
Linux环境基础开发工具——(2)vim
linux·开发语言·经验分享·笔记·vim·学习方法
小郝 小郝37 分钟前
【C语言】strstr查找字符串函数
c语言·开发语言
yinhezhanshen43 分钟前
理解rust里面的copy和clone
开发语言·后端·rust
Jtti1 小时前
PHP在Debian环境上的并发处理能力如何
开发语言·debian·php
时光追逐者1 小时前
在 Blazor 中使用 Chart.js 快速创建数据可视化图表
开发语言·javascript·信息可视化·c#·.net·blazor
独好紫罗兰1 小时前
洛谷题单3-P5718 【深基4.例2】找最小值-python-流程图重构
开发语言·python·算法