Linear index upper triangular matrix

by

The equations going from linear index to (i,j) index are

i = n - 2 - floor(sqrt(-8*k + 4*n*(n-1)-7)/2.0 - 0.5)
j = k + i + 1 - n*(n-1)/2 + (n-i)*((n-i)-1)/2

The inverse operation, from (i,j) index to linear index is

k = (n*(n-1)/2) - (n-i)*((n-i)-1)/2 + j - i - 1

Verify in Python with:

from numpy import triu_indices, sqrt
n = 10
for k in range(n*(n-1)/2):
    i = n - 2 - int(sqrt(-8*k + 4*n*(n-1)-7)/2.0 - 0.5)
    j = k + i + 1 - n*(n-1)/2 + (n-i)*((n-i)-1)/2
    assert np.triu_indices(n, k=1)[0][k] == i
    assert np.triu_indices(n, k=1)[1][k] == j

for i in range(n):
    for j in range(i+1, n):
        k = (n*(n-1)/2) - (n-i)*((n-i)-1)/2 + j - i - 1
        assert triu_indices(n, k=1)[0][k] == i
        assert triu_indices(n, k=1)[1][k] == j

More Related Contents:

  1. how to return numpy.array from boost::python?
  2. NumPy style arrays for C++? [closed]
  3. The question is about array in C++ for more understanding in programming subject
  4. How to access 1 element in 2D dynamic array in C++
  5. I keep getting a segfault every time I run this program, but I cannot understand why
  6. What is array to pointer decay?
  7. When a function has a specific-size array parameter, why is it replaced with a pointer?
  8. How do I find the length of an array?
  9. Array[n] vs Array[10] – Initializing array with variable vs numeric literal
  10. Correct way to work with vector of arrays
  11. c++ array – expression must have a constant value
  12. char* vs std::string in c++ [closed]
  13. C/C++ int[] vs int* (pointers vs. array notation). What is the difference?
  14. C++11: Correct std::array initialization?
  15. What is the difference between std::array and std::vector? When do you use one over other? [duplicate]
  16. Why no variable size array in stack?
  17. Now that we have std::array what uses are left for C-style arrays?
  18. C++ int to byte array
  19. How to make an array with a dynamic size? General usage of dynamic arrays (maybe pointers too)? [closed]
  20. How to print 2D Arrays in C++?
  21. Initializer list for dynamic arrays?
  22. Is int arr[ ] valid C++?
  23. Why does numpy.linalg.solve() offer more precise matrix inversions than numpy.linalg.inv()?
  24. What is the safe way to fill multidimensional array using std::fill?
  25. In C/C++, is char* arrayName[][] a pointer to a pointer to a pointer OR a pointer to a pointer?
  26. Function returning array but main showing garbage [duplicate]
  27. Why do we even need the “delete[]” operator?
  28. Passing numpy arrays in Cython to a C function that requires dynamically allocated arrays
  29. Dereferencing one past the end pointer to array type
  30. How do I find a particular value in an array and return its index?

Leave a Comment