How to get the corresponding logic value based on the sum of vector elements

1 回表示 (過去 30 日間)
I have a vector which is [1,1,2,3,5]; I want to loop through all the elements in the vector to check whether the sum of each elements is equal to my input. After that put the corresponfding index of the possible answer as a reference to a new vector which retrun the logic value. For example, if the inout is 6, the corresponding logic value suppose to be [1,0,0,0,1] or anyother combination. The problem is if the value is bigger than 8 it need the sum of 3 elements in vector. How I can get the same answer when the input need three elements add together?

採用された回答

Andrei Bobrov
Andrei Bobrov 2019 年 5 月 14 日
A = [1,1,2,3,5];
my_input = 5;
out = [];
n = numel(A);
ii = 1:n;
for jj = 1:n
k = nchoosek(ii,jj);
r = sum(reshape(A(k),size(k)),2);
lo = r == my_input;
if any(lo)
p = k(lo,:);
m = size(p,1);
out = [out;accumarray([repmat(1:m,jj,1),p(:)],1,[m,n])];
end
end
  3 件のコメント
Andrei Bobrov
Andrei Bobrov 2019 年 5 月 15 日
編集済み: Andrei Bobrov 2019 年 5 月 15 日
Please read 'help' of the MATLAB about nchoosek, reshape.
nchoosek(v,k) - returns a matrix containing all possible combinations of the elements of vector v taken k at a time.
reshape(A,sz) - reshapes A using the size vector, sz, to define size(B). For example, reshape(A,[2,3]) reshapes A into a 2-by-3 matrix. sz must contain at least 2 elements, and prod(sz) must be the same as numel(A).
Row
r = sum(reshape(A(k),size(k)),2);
we can change on block:
if jj == n
AA = Ak(k).';
else
AA = Ak(k);
end
r = sum(AA,2);
here
Ak = A(:);
Zhou Liangyu
Zhou Liangyu 2019 年 5 月 16 日
Thank you very much.

サインインしてコメントする。

その他の回答 (0 件)

カテゴリ

Help Center および File ExchangeMatrices and Arrays についてさらに検索

タグ

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by