分组连续日期

时间:2013-03-14 15:51:24

标签: c# asp.net linq-to-sql

我有List<DateTime> dates;

我有一个课程:

class NonWorkingDay
{
   public DateTime Start;
   public int Days;
}

我正试图找出一种简洁的方法来对它们进行分组。

public List<NonWorkingDay> GetContiguousDates(List<DateTime> dates)
{

}

注意:如果星期五有一个NWD,而下一个是星期一,则应将它们分组。周末不被考虑。

例如,如果我有

September 3 2013
September 20 2013
September 23 2013
September 24 2013
September 30 2013
October 1  2013

输出结果为:

Start = September 3 2013, Days = 1
Start = September 20 2013, Days = 3 //weekend got skipped
Start = September 30 2013, Days = 2

有没有办法做到这一点(没有一堆计数器变量)并使用.Select或.Where等。

由于

1 个答案:

答案 0 :(得分:18)

因此,我们将从这个通用迭代器函数开始。它接受一个序列和一个谓词,它接受两个项并返回一个布尔值。它将读取来自源的项目,同时项目及其前一项目将根据谓词返回true,下一项目将位于“下一组”中。如果返回false,则前一个组已满,并且下一个组已启动。

public static IEnumerable<IEnumerable<T>> GroupWhile<T>(this IEnumerable<T> source
    , Func<T, T, bool> predicate)
{
    using (var iterator = source.GetEnumerator())
    {
        if (!iterator.MoveNext())
            yield break;

        List<T> currentGroup = new List<T>() { iterator.Current };
        while (iterator.MoveNext())
        {
            if (predicate(currentGroup.Last(), iterator.Current))
                currentGroup.Add(iterator.Current);
            else
            {
                yield return currentGroup;
                currentGroup = new List<T>() { iterator.Current };
            }
        }
        yield return currentGroup;
    }
}

我们还需要这个简单的帮助方法,它根据日期获得下一个工作日。如果你想把假期纳入其中,那么从简单到非常困难,但这就是逻辑的发展方向。

public static DateTime GetNextWorkDay(DateTime date)
{
    DateTime next = date.AddDays(1);
    if (next.DayOfWeek == DayOfWeek.Saturday)
        return next.AddDays(2);
    else if (next.DayOfWeek == DayOfWeek.Sunday)
        return next.AddDays(1);
    else
        return next;
}

现在把它们放在一起。首先我们订购日子。 (如果您确保他们总是有序,您可以删除该部分。)然后我们将连续项目分组,而每个项目是前一个工作日的下一个工作日。

然后,我们需要做的就是将IEnumerable<DateTime>个连续日期转换为NonWorkingDay。为此,开始日期是第一个日期,Days是序列的计数。虽然通常同时使用FirstCount会迭代源序列两次,但我们碰巧知道GroupWhile返回的序列实际上是List,因此迭代它多次不是问题,而Count甚至是O(1)。

public IEnumerable<NonWorkingDay> GetContiguousDates(IEnumerable<DateTime> dates)
{
    return dates.OrderBy(d => d)
            .GroupWhile((previous, next) => GetNextWorkDay(previous).Date == next.Date)
            .Select(group => new NonWorkingDay
                {
                    Start = group.First(),
                    Days = group.Count(),
                });
}