1. 程式人生 > >【NOIP2017提高A組模擬10.7】Repulsed

【NOIP2017提高A組模擬10.7】Repulsed

Description:

小w 心裡的火焰就要被熄滅了。
簡便起見,假設小w 的內心是一棵n -1 條邊,n 個節點的樹。
現在你要在每個節點裡放一些個滅火器,每個節點可以放任意多個。
接下來每個節點都要被分配給一個至多k 條邊遠的滅火器,每個滅火器最多能分配給s 個節點。
至少要多少個滅火器才能讓小w 徹底死亡呢?
n <= 10^5, k <= 20, s <= 10^9

題解:

貪心題都是很好騙分的。

fi,kgi,k分別表示到i這個點,它的子樹中,到它距離為k的滅火器能提供的數量,和需求的點數。

如果某些沒有確定點到i距離已經等於k,顯然一定要滿足,不能滿足就要在i點放滅火器。

如果有滅火器的距離加上未確定點的距離等於k,k-1,顯然在i這裡就要匹配了,不然往上走一格,距離增加了2,就超過了k,這樣是不行的。

Code:

#include<cstdio> 
#define ll long long
#define fo(i, x, y) for(int i = x; i <= y; i ++)
#define fd(i, x, y) for(int i = x; i >= y; i --)
using namespace std;

const int N = 100005;

int n, s, k, x, y, ans;
int final[N], next
[N * 2], to[N * 2], tot; ll f[N][21], g[N][21]; int bz[N]; void link(int x, int y) { next[++ tot] = final[x], to[tot] = y, final[x] = tot; next[++ tot] = final[y], to[tot] = x, final[y] = tot; } void Gao(int x) { if(f[x][0] >= g[x][k]) f[x][0] -= g[x][k], g[x][k] = 0; else g[x
][k] -= f[x][0], f[x][0] = 0; } void dg(int x) { bz[x] = 1; { g[x][0] = 1; for(int i = final[x]; i; i = next[i]) { int y = to[i]; if(bz[y]) continue; dg(y); fo(j, 0, k - 1) f[x][j + 1] += f[y][j], g[x][j + 1] += g[y][j]; } while(g[x][k]) { ans ++; if(g[x][k] <= s) f[x][0] = s - g[x][k], g[x][k] = 0; else g[x][k] -= s; } fo(i, 0, k) { if(f[x][i] >= g[x][k - i]) f[x][i] -= g[x][k - i], g[x][k - i] = 0; else g[x][k - i] -= f[x][i], f[x][i] = 0; } fo(i, 0, k - 1) { if(f[x][i] >= g[x][k - i - 1]) f[x][i] -= g[x][k - i - 1], g[x][k - i - 1] = 0; else g[x][k - i - 1] -= f[x][i], f[x][i] = 0; } } bz[x] = 0; } int main() { freopen("repulsed.in", "r", stdin); freopen("repulsed.out", "w", stdout); scanf("%d %d %d", &n, &s, &k); s = n < s ? n : s; fo(i, 1, n - 1) { scanf("%d %d", &x, &y); link(x, y); } dg(1); fd(i, k, 0) { while(g[1][i]) { fd(j, k - i, 0) { if(f[1][j] >= g[1][i]) { f[1][j] -= g[1][i]; g[1][i] = 0; break; } else { g[1][i] -= f[1][j]; f[1][j] = 0; } } if(g[1][i]) ans ++, f[1][0] += s; } } printf("%d", ans); }