erlang算法系列-leetcode 2166. 设计位集(中等)

难点是需要记录Flip的次数进行异或,使得flip男足O(1)的复杂度。使用到进程字典,array,integer_to_list,erlang:list_to_bitstring。

设计位集-原题

位集 Bitset 是一种能以紧凑形式存储位的数据结构。

请你实现 Bitset 类。

Bitset(int size) 用 size 个位初始化 Bitset ,所有位都是 0 。
void fix(int idx) 将下标为 idx 的位上的值更新为 1 。如果值已经是 1 ,则不会发生任何改变。
void unfix(int idx) 将下标为 idx 的位上的值更新为 0 。如果值已经是 0 ,则不会发生任何改变。
void flip() 翻转 Bitset 中每一位上的值。换句话说,所有值为 0 的位将会变成 1 ,反之亦然。
boolean all() 检查 Bitset 中 每一位 的值是否都是 1 。如果满足此条件,返回 true ;否则,返回 false 。
boolean one() 检查 Bitset 中 是否 至少一位 的值是 1 。如果满足此条件,返回 true ;否则,返回 false 。
int count() 返回 Bitset 中值为 1 的位的 总数 。
String toString() 返回 Bitset 的当前组成情况。注意,在结果字符串中,第 i 个下标处的字符应该与 Bitset 中的第 i 位一致。

示例:

输入
["Bitset", "fix", "fix", "flip", "all", "unfix", "flip", "one", "unfix", "count", "toString"]
[[5], [3], [1], [], [], [0], [], [], [0], [], []]
输出
[null, null, null, null, false, null, null, true, null, 2, "01010"]

解释
Bitset bs = new Bitset(5); // bitset = "00000".
bs.fix(3);     // 将 idx = 3 处的值更新为 1 ,此时 bitset = "00010" 。
bs.fix(1);     // 将 idx = 1 处的值更新为 1 ,此时 bitset = "01010" 。
bs.flip();     // 翻转每一位上的值,此时 bitset = "10101" 。
bs.all();      // 返回 False ,bitset 中的值不全为 1 。
bs.unfix(0);   // 将 idx = 0 处的值更新为 0 ,此时 bitset = "00101" 。
bs.flip();     // 翻转每一位上的值,此时 bitset = "11010" 。
bs.one();      // 返回 True ,至少存在一位的值为 1 。
bs.unfix(0);   // 将 idx = 0 处的值更新为 0 ,此时 bitset = "01010" 。
bs.count();    // 返回 2 ,当前有 2 位的值为 1 。
bs.toString(); // 返回 "01010" ,即 bitset 的当前组成情况。

提示:

1 <= size <= 105
0 <= idx <= size - 1
至多调用 fix、unfix、flip、all、one、count 和 toString 方法 总共 105 次
至少调用 all、one、count 或 toString 方法一次
至多调用 toString 方法 5 次

-spec bitset_init_(Size :: integer()) -> any().
bitset_init_(Size) ->
    Arr = array:new(Size, {default,0}),
    Data = {0, 0, Arr},
    put(bitdata, Data).

-spec bitset_fix(Idx :: integer()) -> any().
bitset_fix(Idx) ->
    {Count, Flip, Arr} = get(bitdata),
    Value = array:get(Idx, Arr),
    case Flip bxor Value of
        0 ->
            Data = {Count+1, Flip, array:set(Idx, Value bxor 1, Arr)},
            put(bitdata, Data);
        _ ->
        ok
    end.

-spec bitset_unfix(Idx :: integer()) -> any().
bitset_unfix(Idx) ->
    {Count, Flip, Arr} = get(bitdata),
    Value = array:get(Idx, Arr),
    case Flip bxor Value of
        1 ->
            Data = {Count-1, Flip, array:set(Idx, Value bxor 1, Arr)},
            put(bitdata, Data);
        _ ->
        ok
    end.   

-spec bitset_flip() -> any().
bitset_flip() ->
    {Count, Flip, Arr} = get(bitdata),
    Size = array:size(Arr),
    Data = {Size - Count, Flip bxor 1 , Arr},
    put(bitdata, Data).    

-spec bitset_all() -> boolean().
bitset_all() ->
    {Count, Flip, Arr} = get(bitdata),
    Size = array:size(Arr),
    Count =:= Size.

-spec bitset_one() -> boolean().
bitset_one() ->
    {Count, Flip, Arr} = get(bitdata),
    Value = array:size(Arr),
    Count > 0.

-spec bitset_count() -> integer().
bitset_count() ->
    {Count, Flip, Arr} = get(bitdata),
    Value = array:size(Arr),
    Count.

-spec bitset_to_string() -> unicode:unicode_binary().
bitset_to_string() ->
    {Count, Flip, Arr} = get(bitdata),
    List = array:to_list(Arr),
    List1 = [ integer_to_list(N bxor Flip) || N <- List],
    erlang:list_to_bitstring(List1).


%% Your functions will be called as such:
%% bitset_init_(Size),
%% bitset_fix(Idx),
%% bitset_unfix(Idx),
%% bitset_flip(),
%% Param_4 = bitset_all(),
%% Param_5 = bitset_one(),
%% Param_6 = bitset_count(),
%% Param_7 = bitset_to_string(),

%% bitset_init_ will be called before every test case, in which you can do some necessary initializations.

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值