1. 程式人生 > >約瑟夫環【dp】

約瑟夫環【dp】

N個人坐成一個圓環(編號為1 - N),從第1個人開始報數,數到K的人出列,後面的人重新從1開始報數。問最後剩下的人的編號。
例如:N =3,K = 2。2號先出列,然後是1號,最後剩下的是3號。

思路:
簡單想法可以直接摸擬,時間複雜度O(nk),資料規模大時會超時。運用遞推思想。

  1. 1~n迴圈,簡化為0~n-1,處理更方便。
  2. 第一次出列時,剩餘k,k+1,…,n-1,0,1,…,k-2,將其對應為0,1,…,n-2,可以看出存在關係f(N) = (f(N-1)+k)%N。
  3. 對任意i,都有f(i) = (f(i-1)+k)%i;
  4. f(0) = 0;
  5. 滾動數字,空間複雜度由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;
}