fix: Correct return statement in fft_convolve function to ensure proper resizing of output vector

weilycoder · weilycoder · commit 30b97a5ad00c · 2025-11-07T09:14:15.000+08:00
diff --git a/weilycoder/poly/fft_convolve.hpp b/weilycoder/poly/fft_convolve.hpp
@@ -29,7 +29,8 @@ std::vector<std::complex<float_t>> fft_convolve(std::vector<std::complex<float_t
   fft(a), fft(b);
   for (size_t i = 0; i < n; ++i)
     a[i] *= b[i];
-  return fft<-1>(a), a;
+  fft<-1>(a), a.resize(a.size() + b.size() - 1);
+  return a;
 }
 
 /**

Original file line number	Diff line number	Diff line change
`@@ -29,7 +29,8 @@ std::vector<std::complex<float_t>> fft_convolve(std::vector<std::complex<float_t`
`29`	`29`	`fft(a), fft(b);`
`30`	`30`	`for (size_t i = 0; i < n; ++i)`
`31`	`31`	`a[i] *= b[i];`
`32`		`- return fft<-1>(a), a;`
	`32`	`+ fft<-1>(a), a.resize(a.size() + b.size() - 1);`
	`33`	`+ return a;`
`33`	`34`	`}`
`34`	`35`
`35`	`36`	`/**`