Dang_Quan_10_Tin

DHBB112021_KIOSKS

Aug 15th, 2021 (edited)
397
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 1.84 KB | None | 0 0
  1. #define task "KIOSKS"
  2. #include <iostream>
  3. #include <cstdio>
  4. #include <vector>
  5. using namespace std;
  6.  
  7. using ll = long long;
  8.  
  9. void read(int &x)
  10. {
  11.     x = 0;
  12.     register int c;
  13.     while ((c = getchar()) && (c > '9' || c < '0'))
  14.         ;
  15.     for (; c >= '0' && c <= '9'; c = getchar())
  16.         x = x * 10 + c - '0';
  17. }
  18.  
  19. constexpr int N = 1e4 + 5;
  20. constexpr ll mod = 1e9 + 7;
  21. ll ans(0);
  22. ll dp[N];
  23. int n, k, a[N], cnt[N], l(0), id[N];
  24. vector<int> adj[N];
  25. bool Sub2(1);
  26.  
  27. void Read()
  28. {
  29.  
  30.     //cin >> n >> k;
  31.     read(n), read(k);
  32.     for (int i = 1; i <= n; ++i)
  33.         //cin >> a[i];
  34.         read(a[i]);
  35.     for (int i = 1; i < n; ++i)
  36.     {
  37.         int u, v;
  38.         //cin >> u >> v;
  39.         read(u), read(v);
  40.         adj[u].emplace_back(v);
  41.         adj[v].emplace_back(u);
  42.         Sub2 = Sub2 && ((int)adj[u].size() <= 2) && ((int)adj[v].size() <= 2);
  43.     }
  44. }
  45.  
  46. #define bit(i, x) (((x) >> (i)) & 1)
  47.  
  48. /// dp[i]: số cách chọn các ki ốt để giữ lại sao cho PHẢI chứa i và không được chứa ki ốt nào bán loại hàng được bật trong mask
  49.  
  50. void dfs_1(int v, int p, int mask, int sign)
  51. {
  52.     dp[v] = 1;
  53.     if (a[v] <= k && bit(a[v] - 1, mask))
  54.         dp[v] = 0;
  55.  
  56.     for (auto i : adj[v])
  57.         if (i != p)
  58.         {
  59.             dfs_1(i, v, mask, sign);
  60.             dp[v] = dp[v] * (dp[i] + 1) % mod;
  61.         }
  62.  
  63.     (ans += sign * dp[v]) %= mod;
  64. }
  65.  
  66. void Sub_3()
  67. {
  68.     /// bao hàm loại trừ
  69.     for (int i = 0; i < (1 << k); ++i)
  70.         dfs_1(1, 0, i, (__builtin_popcount(i) & 1) ? -1 : 1);
  71.     cout << (ans + mod) % mod;
  72. }
  73. int32_t main()
  74. {
  75.     ios_base::sync_with_stdio(0);
  76.     cin.tie(0);
  77.     cout.tie(0);
  78.  
  79.     if (fopen(task ".INP", "r"))
  80.     {
  81.         freopen(task ".INP", "r", stdin);
  82.         freopen(task ".OUT", "w", stdout);
  83.     }
  84.  
  85.     Read();
  86.     Sub_3();
  87. }
Advertisement
Add Comment
Please, Sign In to add comment