如何实现"InterpolatedVector"?
How to implement a "InterpolatedVector"?
我有一个std::vector<double>
,需要对其值进行插值。例如,只有1个中间值,并且给定一个充满的矢量
1 / 2 / 3 / 4
我想访问以下值
1 / 1.5 / 2 / 2.5 / 3 / 3.5 / 4
当然,我不必存储这些中间值(简单的线性插值,我不需要经常读取它们),所以我写了这个简单的类:
typedef std::vector<double> DVector;
class InterpolatedVector {
public:
InterpolatedVector(const DVector& v,int steps) : v(v),steps(steps){}
double at(int i){
int j = i%steps;
int k = (int)i/steps;
if (i==0){return v[0];}
else if (i==v.size()*steps){return v.back();}
else {return ((steps-j)*v[k] + j*v[k+1])/steps;}
}
int size(){return steps*(v.size()-1) + 1;}
private:
DVector v;
int steps;
};
它很好用,我(几乎)得到了我想要的。然而,这个"容器"我不能与std::算法一起使用,我也没有迭代器。
我应该如何实现这个"插值向量",以便我可以执行类似的操作
std::accumulate(/* passing Interpolated iterators? */ );
考虑到您已经有了处理索引本身的代码,将其包装为迭代器非常容易。如果你能原谅我,我也会让它更通用一点。
#include <vector>
#include <iterator>
template <class T>
class InterpolatedVector {
typedef std::vector<T> DVector;
public:
InterpolatedVector(const DVector& v,int steps) : v(v),steps(steps){}
T at(int i){
int j = i%steps;
int k = (int)i/steps;
if (i==0){return v[0];}
else if (i==v.size()*steps){return v.back();}
else {return ((steps-j)*v[k] + j*v[k+1])/steps;}
}
int size(){return steps*(v.size()-1) + 1;}
class iterator : public std::iterator < std::random_access_iterator_tag, T > {
InterpolatedVector *vec;
int index;
public:
iterator(InterpolatedVector &d, int index) : vec(&d), index(index) {}
iterator &operator++() { ++index; return *this; }
iterator operator++(int) { iterator tmp{ *vec, index }; ++index; return tmp; }
iterator operator+(int off) { return iterator(*vec, index + off); }
iterator operator-(int off) { return iterator(*vec, index - off); }
value_type operator*() { return (*vec).at(index); }
bool operator!=(iterator const &other) { return index != other.index; }
bool operator<(iterator const &other) { return index < other.index; }
};
iterator begin() { return iterator(*this, 0); }
iterator end() { return iterator(*this, size()); }
private:
DVector v;
int steps;
};
和一个快速的演示代码来测试它:
#include <iostream>
int main() {
std::vector<double> d{ 1, 2, 3, 4 };
InterpolatedVector<double> id(d, 2);
std::copy(id.begin(), id.end(), std::ostream_iterator<double>(std::cout, "t"));
std::cout << "n";
std::vector<int> i{ 0, 5 };
InterpolatedVector<int> ii(i, 5);
std::copy(ii.begin(), ii.end(), std::ostream_iterator<int>(std::cout, "t"));
}
输出:
1 1.5 2 2.5 3 3.5 4
0 1 2 3 4 5
当然,有些算法仍然无法对这种"集合"做太多的处理。尝试将插值集合提供给std::sort
没有多大意义(您显然需要对底层容器进行排序)。只要算法只需要读取数据,这应该没问题。
如果您想将其与算法一起使用,请用中间值填充向量。这会简单得多。
相关文章:
- 没有找到相关文章