NSCountedSet(NSDictionaryExtensions)

NSCountedSet+Dictionary.h

#import <Foundation/Foundation.h>

@interface NSCountedSet (NSDictionaryExtensions)

- (id)initWithDictionary:(NSDictionary *)dict;
- (void)addObjectsFromDictionary:(NSDictionary *)dict;
- (NSDictionary *)dictionary;

@end

NSCountedSet+Dictionary.m

#import "NSCountedSet+Dictionary.h"

@implementation NSCountedSet (NSDictionaryExtensions)

- (id)initWithDictionary:(NSDictionary *)dict
{
	if ((self = [super initWithCapacity:[dict count]]))
	{
		[self addObjectsFromDictionary:dict];
	}

	return self;
}

- (void)addObjectsFromDictionary:(NSDictionary *)dict
{
	[dict enumerateKeysAndObjectsUsingBlock:^(id key, id obj, BOOL *stop){
		NSUInteger count = [obj unsignedIntegerValue];
		for (NSUInteger i = 0; i < count; ++i)
		{
			[self addObject:key];
		}
	}];	
}

- (NSDictionary *)dictionary
{
	NSMutableDictionary *dict = [NSMutableDictionary dictionaryWithCapacity:self.count];
	
	for (id object in self)
	{
		NSUInteger count = [self countForObject:object];
		[dict setObject:[NSNumber numberWithUnsignedInteger:count] forKey:object];
	}

	return dict;
}

@end