zoukankan      html  css  js  c++  java
  • 代码片段

    ------------------------------------------------------------
    如果用于多例程,可以使用下面的版本:
    ------------------------------------------------------------
    
    package main
    
    import (
    	"fmt"
    	"sort"
    	"sync"
    )
    
    type Set struct {
    	sync.RWMutex
    	m map[int]bool
    }
    
    // 新建集合对象
    // 可以传入初始元素
    func New(items ...int) *Set {
    	s := &Set{
    		m: make(map[int]bool, len(items)),
    	}
    	s.Add(items...)
    	return s
    }
    
    // 创建副本
    func (s *Set) Duplicate() *Set {
    	s.Lock()
    	defer s.Unlock()
    	r := &Set{
    		m: make(map[int]bool, len(s.m)),
    	}
    	for e := range s.m {
    		r.m[e] = true
    	}
    	return r
    }
    
    // 添加元素
    func (s *Set) Add(items ...int) {
    	s.Lock()
    	defer s.Unlock()
    	for _, v := range items {
    		s.m[v] = true
    	}
    }
    
    // 删除元素
    func (s *Set) Remove(items ...int) {
    	s.Lock()
    	defer s.Unlock()
    	for _, v := range items {
    		delete(s.m, v)
    	}
    }
    
    // 判断元素是否存在
    func (s *Set) Has(items ...int) bool {
    	s.RLock()
    	defer s.RUnlock()
    	for _, v := range items {
    		if _, ok := s.m[v]; !ok {
    			return false
    		}
    	}
    	return true
    }
    
    // 统计元素个数
    func (s *Set) Count() int {
    	s.Lock()
    	defer s.Unlock()
    	return len(s.m)
    }
    
    // 清空集合
    func (s *Set) Clear() {
    	s.Lock()
    	defer s.Unlock()
    	s.m = map[int]bool{}
    }
    
    // 空集合判断
    func (s *Set) Empty() bool {
    	s.Lock()
    	defer s.Unlock()
    	return len(s.m) == 0
    }
    
    // 获取元素列表(无序)
    func (s *Set) List() []int {
    	s.RLock()
    	defer s.RUnlock()
    	list := make([]int, 0, len(s.m))
    	for item := range s.m {
    		list = append(list, item)
    	}
    	return list
    }
    
    // 获取元素列表(有序)
    func (s *Set) SortedList() []int {
    	s.RLock()
    	defer s.RUnlock()
    	list := make([]int, 0, len(s.m))
    	for item := range s.m {
    		list = append(list, item)
    	}
    	sort.Ints(list)
    	return list
    }
    
    // 并集
    // 获取 s 与参数的并集,结果存入 s
    func (s *Set) Union(sets ...*Set) {
    	// 为了防止多例程死锁,不能同时锁定两个集合
    	// 所以这里没有锁定 s,而是创建了一个临时集合
    	r := s.Duplicate()
    	// 获取并集
    	for _, set := range sets {
    		set.Lock()
    		for e := range set.m {
    			r.m[e] = true
    		}
    		set.Unlock()
    	}
    	// 将结果转入 s
    	s.Lock()
    	defer s.Unlock()
    	s.m = map[int]bool{}
    	for e := range r.m {
    		s.m[e] = true
    	}
    }
    
    // 并集(函数)
    // 获取所有参数的并集,并返回
    func Union(sets ...*Set) *Set {
    	// 处理参数数量
    	if len(sets) == 0 {
    		return New()
    	} else if len(sets) == 1 {
    		return sets[0]
    	}
    	// 获取并集
    	r := sets[0].Duplicate()
    	for _, set := range sets[1:] {
    		set.Lock()
    		for e := range set.m {
    			r.m[e] = true
    		}
    		set.Unlock()
    	}
    	return r
    }
    
    // 差集
    // 获取 s 与所有参数的差集,结果存入 s
    func (s *Set) Minus(sets ...*Set) {
    	// 为了防止多例程死锁,不能同时锁定两个集合
    	// 所以这里没有锁定 s,而是创建了一个临时集合
    	r := s.Duplicate()
    	// 获取差集
    	for _, set := range sets {
    		set.Lock()
    		for e := range set.m {
    			delete(r.m, e)
    		}
    		set.Unlock()
    	}
    	// 将结果转入 s
    	s.Lock()
    	defer s.Unlock()
    	s.m = map[int]bool{}
    	for e := range r.m {
    		s.m[e] = true
    	}
    }
    
    // 差集(函数)
    // 获取第 1 个参数与其它参数的差集,并返回
    func Minus(sets ...*Set) *Set {
    	// 处理参数数量
    	if len(sets) == 0 {
    		return New()
    	} else if len(sets) == 1 {
    		return sets[0]
    	}
    	// 获取差集
    	r := sets[0].Duplicate()
    	for _, set := range sets[1:] {
    		for e := range set.m {
    			delete(r.m, e)
    		}
    	}
    	return r
    }
    
    // 交集
    // 获取 s 与其它参数的交集,结果存入 s
    func (s *Set) Intersect(sets ...*Set) {
    	// 为了防止多例程死锁,不能同时锁定两个集合
    	// 所以这里没有锁定 s,而是创建了一个临时集合
    	r := s.Duplicate()
    	// 获取交集
    	for _, set := range sets {
    		set.Lock()
    		for e := range r.m {
    			if _, ok := set.m[e]; !ok {
    				delete(r.m, e)
    			}
    		}
    		set.Unlock()
    	}
    	// 将结果转入 s
    	s.Lock()
    	defer s.Unlock()
    	s.m = map[int]bool{}
    	for e := range r.m {
    		s.m[e] = true
    	}
    }
    
    // 交集(函数)
    // 获取所有参数的交集,并返回
    func Intersect(sets ...*Set) *Set {
    	// 处理参数数量
    	if len(sets) == 0 {
    		return New()
    	} else if len(sets) == 1 {
    		return sets[0]
    	}
    	// 获取交集
    	r := sets[0].Duplicate()
    	for _, set := range sets[1:] {
    		for e := range r.m {
    			if _, ok := set.m[e]; !ok {
    				delete(r.m, e)
    			}
    		}
    	}
    	return r
    }
    
    // 补集
    // 获取 s 相对于 full 的补集,结果存入 s
    func (s *Set) Complement(full *Set) {
    	r := full.Duplicate()
    	s.Lock()
    	defer s.Unlock()
    	// 获取补集
    	for e := range s.m {
    		delete(r.m, e)
    	}
    	// 将结果转入 s
    	s.m = map[int]bool{}
    	for e := range r.m {
    		s.m[e] = true
    	}
    }
    
    // 补集(函数)
    // 获取 sub 相对于 full 的补集,并返回
    func Complement(sub, full *Set) *Set {
    	r := full.Duplicate()
    	sub.Lock()
    	defer sub.Unlock()
    	for e := range sub.m {
    		delete(r.m, e)
    	}
    	return r
    }
    
    func main() {
    	s1 := New(1, 2, 3, 4, 5, 6, 7, 8)
    	s2 := New(3, 4, 5, 6)
    	s3 := New(1, 2, 5, 6, 8, 9)
    	
    	// 创建 10 个 goroutine 同步操作 s2,看会不会死锁
    	wg := sync.WaitGroup{}
    	for i := 0; i < 10; i++ {
    		wg.Add(1)
    		go func(n int) {
    			for i := 0; i < 100; i++ {
    				s2.Union(s1) // 获取并集
    				fmt.Printf("%2v:s2 + %v = %v
    ", n, s1.SortedList(), s2.SortedList())
    
    				s2.Minus(s3) // 获取差集
    				fmt.Printf("%2v:s2 - %v = %v
    ", n, s3.SortedList(), s2.SortedList())
    
    				s2.Intersect(s1) // 获取交集
    				fmt.Printf("%2v:s2 * %v = %v
    ", n, s1.SortedList(), s2.SortedList())
    
    				s2.Complement(s1) // 获取 s2 相对于 s1 的补集
    				fmt.Printf("%2v:%v / s2 = %v
    ", n, s1.SortedList(), s2.SortedList())
    			}
    			wg.Done()
    		}(i)
    	}
    	wg.Wait()
    }
    
    ------------------------------------------------------------
    如果不用于多例程,可以使用下面的简单版本:
    ------------------------------------------------------------
    
    package main
    
    import (
    	"fmt"
    	"sort"
    )
    
    type Set map[int]bool
    
    // 新建集合对象
    // 可以传入初始元素
    func New(items ...int) Set {
    	s := make(Set, len(items))
    	s.Add(items...)
    	return s
    }
    
    // 创建副本
    func (s Set) Duplicate() Set {
    	r := make(map[int]bool, len(s))
    	for e := range s {
    		r[e] = true
    	}
    	return r
    }
    
    // 添加元素
    func (s Set) Add(items ...int) {
    	for _, v := range items {
    		s[v] = true
    	}
    }
    
    // 删除元素
    func (s Set) Remove(items ...int) {
    	for _, v := range items {
    		delete(s, v)
    	}
    }
    
    // 判断元素是否存在
    func (s Set) Has(items ...int) bool {
    	for _, v := range items {
    		if _, ok := s[v]; !ok {
    			return false
    		}
    	}
    	return true
    }
    
    // 统计元素个数
    func (s Set) Count() int {
    	return len(s)
    }
    
    // 清空集合
    func (s Set) Clear() {
    	s = map[int]bool{}
    }
    
    // 空集合判断
    func (s Set) Empty() bool {
    	return len(s) == 0
    }
    
    // 获取元素列表(无序)
    func (s Set) List() []int {
    	list := make([]int, 0, len(s))
    	for item := range s {
    		list = append(list, item)
    	}
    	return list
    }
    
    // 获取元素列表(有序)
    func (s Set) SortedList() []int {
    	list := s.List()
    	sort.Ints(list)
    	return list
    }
    
    // 并集
    // 获取 s 与参数的并集,结果存入 s
    func (s Set) Union(sets ...Set) {
    	for _, set := range sets {
    		for e := range set {
    			s[e] = true
    		}
    	}
    }
    
    // 并集(函数)
    // 获取所有参数的并集,并返回
    func Union(sets ...Set) Set {
    	// 处理参数数量
    	if len(sets) == 0 {
    		return New()
    	} else if len(sets) == 1 {
    		return sets[0]
    	}
    	// 获取并集
    	r := sets[0].Duplicate()
    	for _, set := range sets[1:] {
    		for e := range set {
    			r[e] = true
    		}
    	}
    	return r
    }
    
    // 差集
    // 获取 s 与所有参数的差集,结果存入 s
    func (s Set) Minus(sets ...Set) {
    	for _, set := range sets {
    		for e := range set {
    			delete(s, e)
    		}
    	}
    }
    
    // 差集(函数)
    // 获取第 1 个参数与其它参数的差集,并返回
    func Minus(sets ...Set) Set {
    	// 处理参数数量
    	if len(sets) == 0 {
    		return New()
    	} else if len(sets) == 1 {
    		return sets[0]
    	}
    	// 获取差集
    	r := sets[0].Duplicate()
    	for _, set := range sets[1:] {
    		for e := range set {
    			delete(r, e)
    		}
    	}
    	return r
    }
    
    // 交集
    // 获取 s 与其它参数的交集,结果存入 s
    func (s Set) Intersect(sets ...Set) {
    	for _, set := range sets {
    		for e := range s {
    			if _, ok := set[e]; !ok {
    				delete(s, e)
    			}
    		}
    	}
    }
    
    // 交集(函数)
    // 获取所有参数的交集,并返回
    func Intersect(sets ...Set) Set {
    	// 处理参数数量
    	if len(sets) == 0 {
    		return New()
    	} else if len(sets) == 1 {
    		return sets[0]
    	}
    	// 获取交集
    	r := sets[0].Duplicate()
    	for _, set := range sets[1:] {
    		for e := range r {
    			if _, ok := set[e]; !ok {
    				delete(r, e)
    			}
    		}
    	}
    	return r
    }
    
    // 补集
    // 获取 s 相对于 full 的补集,结果存入 s
    func (s Set) Complement(full Set) {
    	r := s.Duplicate()
    	s.Clear()
    	for e := range full {
    		if _, ok := r[e]; !ok {
    			s[e] = true
    		}
    	}
    }
    
    // 补集(函数)
    // 获取 sub 相对于 full 的补集,并返回
    func Complement(sub, full Set) Set {
    	r := full.Duplicate()
    	for e := range sub {
    		delete(r, e)
    	}
    	return r
    }
    
    func main() {
    	s1 := New(1, 2, 3, 4, 5, 6, 7, 8)
    	s2 := New(3, 4, 5, 6)
    	s3 := New(5, 6, 8, 9)
    	r1 := Union(s1, s2, s3)     // 获取并集
    	r2 := Minus(s1, s2, s3)     // 获取差集
    	r3 := Intersect(s1, s2, s3) // 获取交集
    	r4 := Complement(s2, s1)    // 获取 s2 相对于 s1 的补集
    	fmt.Println(r1.SortedList())
    	fmt.Println(r2.SortedList())
    	fmt.Println(r3.SortedList())
    	fmt.Println(r4.SortedList())
    }
    
    
    
    
  • 相关阅读:
    如何创建一个基于 .NET Core 3 的 WPF 项目
    阅读源码系列
    近期计划
    Redis 源码走读(二)对象系统
    Redis 源码走读(一)事件驱动机制与命令处理
    basic paxos解析
    ThreadLocal深度解析
    MySQL InnoDB MVCC深度分析
    java泛型详解
    内存可见性,指令重排序,JIT。。。。。。从一个知乎问题谈起
  • 原文地址:https://www.cnblogs.com/golove/p/5840145.html
Copyright © 2011-2022 走看看