partition.h

Go to the documentation of this file.

// -*- C++ -*-

#if !defined(__numerical_partition_h__)
#define __numerical_partition_h__

#include "defs.h"

#include <algorithm>

#include <cassert>

BEGIN_NAMESPACE_NUMERICAL


inline
int
partition(const int x, const int n, const int i) {
#ifdef DEBUG_numerical
  assert(x >= 0 && n > 0 && 0 <= i && i < n);
#endif
  int p = x / n;
  if (i < x % n) {
    ++p;
  }
  return p;
}


inline
void
partitionRange(const int x, const int n, const int i, int* a, int* b) {
  const int p = x / n;
  *a = p * i;
  *a += std::min(i, x % n);
  *b = *a + partition(x, n, i);
}

END_NAMESPACE_NUMERICAL

#endif

---