Advertisement
Guest User

Untitled

a guest
Jan 18th, 2019
150
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 3.13 KB | None | 0 0
  1. #include <iostream>
  2. #include <cstdio>
  3. #include <cstdlib>
  4. #include <algorithm>
  5. #include <cmath>
  6. #include <vector>
  7. #include <set>
  8. #include <map>
  9. #include <unordered_set>
  10. #include <unordered_map>
  11. #include <queue>
  12. #include <ctime>
  13. #include <cassert>
  14. #include <complex>
  15. #include <string>
  16. #include <cstring>
  17. #include <chrono>
  18. #include <random>
  19. #include <queue>
  20. #include <bitset>
  21. using namespace std;
  22.  
  23. #ifdef LOCAL
  24.     #define eprintf(...) fprintf(stderr, __VA_ARGS__)
  25. #else
  26.     #define eprintf(...) 42
  27. #endif
  28.  
  29. typedef long long ll;
  30. typedef pair<int, int> pii;
  31. typedef pair<ll, int> pli;
  32. typedef pair<ll, ll> pll;
  33. typedef long double ld;
  34. #define mp make_pair
  35. mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
  36.  
  37. const ll MOD = 1LL << 32;
  38. ll add(ll x, ll y) {
  39.     x += y;
  40.     if (x >= MOD) return x - MOD;
  41.     return x;
  42. }
  43. ll mult(ll x, ll y) {
  44.     return (x * y) % MOD;
  45. }
  46.  
  47. struct Item {
  48.     int w;
  49.     int id, pos;
  50.  
  51.     Item() : w(), id(), pos() {}
  52.     Item(int _w, int _id, int _pos) : w(_w), id(_id), pos(_pos) {}
  53.  
  54.     bool operator < (const Item &I) const {
  55.         if (w != I.w) return w < I.w;
  56.         if (id != I.id) return id < I.id;
  57.         return pos < I.pos;
  58.     }
  59. };
  60.  
  61. const int N = 1010;
  62. map<Item, int> a;
  63. vector<int> g[N];
  64. int ed[2 * N][3];
  65. int n, m;
  66. vector<int> b[N];
  67. int k;
  68. ll ans;
  69. int par[N];
  70. int h[N];
  71. int sumAll = 0;
  72.  
  73. int getOther(int id, int v) {
  74.     return ed[id][0] ^ ed[id][1] ^ v;
  75. }
  76.  
  77. void dfs(int v) {
  78.     for (int id : g[v]) {
  79.         int u = getOther(id, v);
  80.         if (par[u] == -1) {
  81.             par[u] = id;
  82.             h[u] = h[v] + 1;
  83.             dfs(u);
  84.         } else if (h[u] < h[v] - 1) {
  85.             b[m].push_back(ed[id][2]);
  86.             int w = v;
  87.             while(w != u) {
  88.                 b[m].push_back(ed[par[w]][2]);
  89.                 w = getOther(par[w], w);
  90.             }
  91.             m++;
  92.         }
  93.     }
  94. }
  95.  
  96. int main()
  97. {
  98. //  freopen("input.txt", "r", stdin);
  99. //  freopen("output.txt", "w", stdout);
  100.  
  101.     scanf("%d%d", &n, &m);
  102.     for (int i = 0; i < m; i++) {
  103.         scanf("%d%d%d", &ed[i][0], &ed[i][1], &ed[i][2]);
  104.         ed[i][0]--;
  105.         ed[i][1]--;
  106.         g[ed[i][0]].push_back(i);
  107.         g[ed[i][1]].push_back(i);
  108.         sumAll += ed[i][2];
  109.     }
  110.     scanf("%d", &k);
  111.     m = 0;
  112.     for (int i = 0; i < n; i++)
  113.         par[i] = -1;
  114.     par[0] = -2;
  115.     dfs(0);
  116.     for (int id = 0; id < m; id++) {
  117.         sort(b[id].begin(), b[id].end());
  118.         reverse(b[id].begin(), b[id].end());
  119.         sumAll -= b[id][0];
  120.         for (int i = 0; i < (int)b[id].size() - 1; i++)
  121.             b[id][i] = b[id][i] - b[id][i + 1];
  122.         b[id].pop_back();
  123.     }
  124.     sort(b, b + m);
  125.     a[Item(sumAll, 0, 0)] = 1;
  126.     ll curNum = 1;
  127.     while(!a.empty() && k > 0) {
  128.         Item v = a.begin()->first;
  129.         int cnt = a.begin()->second;
  130.         a.erase(a.begin());
  131.         for (int i = 0; i < cnt && k > 0; i++) {
  132.             ans = add(ans, mult(curNum, v.w));
  133.             curNum++;
  134.             k--;
  135.         }
  136.         if (k == 0) break;
  137.         if ((int)b[v.id].size() > v.pos) {
  138.             Item I = v;
  139.             I.w += b[v.id][v.pos];
  140.             I.pos++;
  141.             a[I] = min(a[I] + cnt, k);
  142.         }
  143.         if (v.id != 0 && v.id < m - 1 && v.pos == 1) {
  144.             Item I = v;
  145.             I.id++;
  146.             I.w += b[I.id][0] - b[v.id][0];
  147.             a[I] = min(a[I] + cnt, k);
  148.         }
  149.         if (v.id < m - 1) {
  150.             Item I = v;
  151.             I.id++;
  152.             I.pos = 1;
  153.             I.w += b[I.id][0];
  154.             a[I] = min(a[I] + cnt, k);
  155.         }
  156.     }
  157.     printf("%lld\n", ans);
  158.  
  159.     return 0;
  160. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement