Re: [討論] 簡單的計算想不到暴力解之外的方法

看板MATLAB作者celestialgod (天)時間10年前 (2015/09/05 11:41)推噓6(6推 0噓 13→)

留言19則, 2人參與討論串2/3 (看更多)

%% function rle % to comput the lengths and values along a vector function [len, values] = rle(x) i = [find(x(2:end) ~= x(1:end-1)); length(x)]; len = diff([0; i]); if nargout > 1 values = x(i); end end %% function inverse_rle % produce a vector with lengths and coresponding values function out = inverse_rle(len, values) res = arrayfun(@(x, y) y*ones(x, 1), len, values, 'UniformOutput', false); out = cat(1, res{:}); end %% main % data generation N = 10000; occurPoints = unique(randi(N, 100, 1)); if mod(length(occurPoints), 2) == 1 occurPoints = [occurPoints; N]; end dat = zeros(N, 1); for i = 1:2:length(occurPoints) dat((occurPoints(i)+1):occurPoints(i+1)) = 1; end %% data generation - method 2 % len = diff([0; occurPoints; N]); % values = repmat(0:1, 1, ceil(length(len)/2)); % values = values(1:length(len))'; % v = inverse_rle(len, values); % isequal(v, dat) % 1 % part 1 [len, values] = rle(dat); valuesCusum = cumsum(values); values(values==1) = valuesCusum(values==1); out_part1 = inverse_rle(len, values); % part 2 [len, values] = rle(dat); tmp = values(values == 0); tmp(len(values==0) < 100) = 1; values(values == 0) = tmp; outTmp = inverse_rle(len, values); [len2, values2] = rle(outTmp); valuesCusum = cumsum(values2); values2(values2==1) = valuesCusum(values2==1); out_part2 = inverse_rle(len2, values2); % part 2 - method 2 (faster, about 2 times) [len, values] = rle(dat); [valuesLen, valuesValues] = rle(values); tmp = [0; cumsum(valuesLen)]; valuesLenNew = zeros(length(valuesValues), 1); k = 1; for i = 1:(length(tmp)-1) valuesLenNew(i) = sum(len((tmp(i)+1):tmp(i+1))); end valuesCusum = cumsum(valuesValues); valuesValues(valuesValues==1) = valuesCusum(valuesValues==1); out_part2_2 = inverse_rle(valuesLenNew, valuesValues); isequal(out_part2, out_part2_2) % 1 第三部分就利用類似第二部分的概念去寫吧，當作練習吧XDD ※ 引述《JamesChen ( )》之銘言： : 問題很簡單，分兩個部分 : 一串 0 1 數列 : 大致長得像是 0 0 0 0 0 1 1 1 1 1 1 1 1 0 0 0 0 0 0 0 0 1 1 1 1 0 0 0 : 簡單來說 1 代表 A 事件發生 0 代表沒有然後這是每個時間點的紀錄 : 所以 A 一旦發生不會只發生 1 個點就結束一定是一串 : 我想要做的是創一個新的數列第 N 次一串 1 變成一串 N : 以上面的例子來說就是後面 4 個 1 都變成 2 : 我只想到 for loop + if 硬幹的方法 : 但實際資料很長又有好多受試者感覺很耗時間 : 第二個部份是 : 如果兩串 1 之前的 0 少於 200 個需要把這兩串 1 合併 (中間的 0 都當作 1) : 我一樣只想到硬解 : 我猜是小弟我不夠熟 Matlab 平常都在用一些自己習慣的 function : 沒有做過類似的事情但應該都有速解 : 希望高手可以幫個忙 : 甚至提點一下就好 -- ※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 123.205.27.107 ※ 文章網址: https://www.ptt.cc/bbs/MATLAB/M.1441424461.A.57D.html

推

JamesChen

09/05 11:53, , 1^F

09/05 11:53, 1^F

第二部分可以再加速，直接用values跟lens做新的lens跟values 不過這部分要再想一想怎麼做XD (已新增)

推

Re: [討論] 簡單的計算 想不到暴力解之外的方法

Re: [討論] 簡單的計算想不到暴力解之外的方法