問：“帶指示器的R”和“帶for的c ++”之間的速度

Question

我覺得以下R代碼的速度很好。 有什么方法可以使用C ++提高速度嗎？ 我覺得我的C ++代碼沒有那么快。

#R-code
> s<-5
> t<-2
> y<-c(1,2,3,4,5)
> r<-c(1,5,5,3,3)
> 
> sindex<-r[r==s]
> tindex<-r[r==t]
> 
> 
> res<-sum(y[sindex])+sum(y[tindex])
> 
> sindex
[1] 5 5
> tindex
numeric(0)
> 
> res
[1] 10

#c++
res1=0; res2=0;
for(i=0;i<n;i++){
if(r[i]==s){
    res1=res1+y[s];
}
if(r[i]==t){
    res2=res2+y[t];
}
}
res=res1+res2

Answer 1

R中的矢量化（sum是矢量化函數）在后台運行C。 通常足夠快...

Answer 2

您可以嘗試以下方法：

res = 0;
for (i = 0; i < n; i++)
     if (r[i] == s || r[i] == t)
          res += y[i];   // y[s] or y[t] seems incorrect

不要期望它會更快。