約瑟夫環【dp】
阿新 • • 發佈:2019-02-10
N個人坐成一個圓環(編號為1 - N),從第1個人開始報數,數到K的人出列,後面的人重新從1開始報數。問最後剩下的人的編號。
例如:N =3,K = 2。2號先出列,然後是1號,最後剩下的是3號。
思路:
簡單想法可以直接摸擬,時間複雜度O(nk),資料規模大時會超時。運用遞推思想。
- 1~n迴圈,簡化為0~n-1,處理更方便。
- 第一次出列時,剩餘k,k+1,…,n-1,0,1,…,k-2,將其對應為0,1,…,n-2,可以看出存在關係f(N) = (f(N-1)+k)%N。
- 對任意i,都有f(i) = (f(i-1)+k)%i;
- f(0) = 0;
- 滾動數字,空間複雜度由O(n)降到O(1)
#include <iostream>
using namespace std;
int main(int argc, char const *argv[])
{
int n, k;
cin >> n >> k;
int ans = 0;
for(int i = 2; i <= n; i++)
ans = (ans + k) % i;
cout << ans + 1 << endl;
return 0;
}