Advertisement
tien_noob

SUBSEQ (Dãy tổng chia hết cho k)

Feb 11th, 2021 (edited)
103
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 1.47 KB | None | 0 0
  1. #include <iostream>
  2. #include <vector>
  3. #include <algorithm>
  4. #include <numeric>
  5. const int N = 1e3;
  6. using namespace std;
  7. int a[N+1], dp[N+1][51], n, k, sum = 0;
  8. void read()
  9. {
  10.     cin >> n >> k;
  11.     for (int i = 1; i <= n; ++ i)
  12.     {
  13.         cin >> a[i];
  14.         a[i] = a[i] % k;
  15.         sum = (sum + a[i])%k;
  16.     }
  17. }
  18. void debug()
  19. {
  20.     for (int i = 0; i <= n; ++ i)
  21.     {
  22.         for (int j = 0; j < k; ++ j)
  23.         {
  24.             if (dp[i][j] >= 1e9)
  25.             {
  26.                 cout << -1;
  27.             }
  28.             else
  29.             {
  30.                 cout << dp[i][j];
  31.             }
  32.             cout << ' ';
  33.         }
  34.         cout <<a[i] << '\n';
  35.     }
  36.     cout << sum;
  37. }
  38. void traceback()
  39. {
  40.     cout << n - dp[n][sum] << '\n';
  41.     for (int i = n; i >= 1; -- i)
  42.     {
  43.         if (dp[i-1][sum] == dp[i][sum])
  44.         {
  45.             cout << i <<' ';
  46.         }
  47.         else
  48.         {
  49.             sum = (sum - a[i] + k)%k;
  50.         }
  51.     }
  52. }
  53. void solve()
  54. {
  55.     dp[0][0] = 0;
  56.     for (int i = 1; i < k; ++ i)
  57.     {
  58.         dp[0][i] = 1e9;
  59.     }
  60.     for (int i = 1; i <= n; ++ i)
  61.     {
  62.         for (int j = 0; j < k; ++ j)
  63.         {
  64.             dp[i][j] = min(dp[i-1][j], dp[i-1][(j - a[i] + k) % k] + 1);
  65.         }
  66.     }
  67.     //debug();
  68.     traceback();
  69. }
  70. int main()
  71. {
  72.     ios_base::sync_with_stdio(false);
  73.     cin.tie(nullptr);
  74.     read();
  75.     solve();
  76. }
  77. // Chuyển đề sang tìm số số nhỏ nhất sao cho tổng lại % k = sum % k
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement