Educational Codeforces Round 54 (Rated for Div.2)

Educational Codeforces Round 54 (Rated for Div.2)

---

D. Edge Deletion

题意：一张n个点的无向图，保留其中k条边，使得有尽可能多的点与1的最短路长度不变。

做法：求出最短路树，然后自底向上删边即可。

#include <bits/stdc++.h>
#define pb push_back
#define P pair<ll,int>
typedef long long ll;
const ll inf = 1e18;
const int N = 3e5 + 7;
using namespace std;
int n, m , k;
struct edge{
    int e,nxt,id; ll w;
}E[N<<1],E2[N<<1];
int h[N], cc, h2[N],cc1;
void add(int u,int v,ll w,int d) {
    E[cc].e = v; E[cc].w = w; E[cc].id = d;
    E[cc].nxt = h[u]; h[u] = cc; ++cc;
}
void add2(int u,int v,ll w,int d) {
    E2[cc1].e = v; E2[cc1].w = w; E2[cc1].id = d;
    E2[cc1].nxt = h2[u]; h2[u] = cc1; ++cc1;
}
struct node{
    int x;ll d;
    node(){}node(int a,ll b){x=a;d=b;}
    bool operator < (const node a)const {
        return a.d < d;
    }
};
ll dis[N];
int vis[N], fa[N], fr[N];
void dij() {
    for(int i=1;i<=n;++i)dis[i]=inf;
    priority_queue<node> q;
    q.push(node(1,0));
    dis[1]=0; fa[1] = 0; fr[1] = -1;
    while(!q.empty()) {
        node tmp = q.top(); q.pop();
        int u=tmp.x;
        if(vis[u])continue;
        vis[u]=1;
        for(int i=h[u];~i;i=E[i].nxt) {
            int v=E[i].e;
            if(dis[v]>dis[u]+E[i].w) {
                dis[v]=dis[u]+E[i].w;
                fa[v] = u;
                fr[v] = E[i].id;
                q.push(node(v,dis[v]));
            }
        }
    }
    return;
}
struct node2{
    int u,v,id; ll w;
    node2(){}
    node2(int a,int b,ll c, int d) {
        u=a; v = b; w = c; id = d;
    }
};
node2 A[N];

int dep[N];
void bfs() {
    queue<int> q;
    memset(dep,-1,sizeof(dep));
    q.push(1); dep[1] = 0;
    while(!q.empty()) {
        int u = q.front(); q.pop();
        for(int i = h2[u]; ~i ; i = E2[i].nxt) {
            int v = E2[i].e;
            if(dep[v] == -1) {
                dep[v] = dep[u] + 1;
                q.push(v);
            }
        }
    }
}

vector< P > B;
int vis2[N];
int main() {
    scanf("%d%d%d",&n,&m,&k);
    memset(h,-1,sizeof(h));
    memset(h2,-1,sizeof(h2));
    for(int i = 1; i <= m; ++i) { int u,v; ll w;
        scanf("%d%d%lld",&u,&v,&w);
        A[i] = node2(u,v,w,i);
        add(u,v,w,i); add(v,u,w,i);
    }
    dij();
    for(int i = 2; i <= n; ++i) {
        int p = fr[i];
        vis2[p] = 1;
        add2(A[p].u,A[p].v,A[p].w,A[p].id);
        add2(A[p].v,A[p].u,A[p].w,A[p].id);
    }
    int e = n-1;
    bfs();
    for(int i = 2; i <= n; ++i) B.pb(P(dep[i],i));
    sort(B.begin(),B.end());
    for(int i = (int)B.size()-1; i >= 0; --i) {
        if(e > k) {
            vis2[fr[B[i].second]] = 0;
            --e;
        }
    }
    printf("%d
",e);
    for(int i = 1; i <= m; ++i) if(vis2[i]) printf("%d ",i); puts("");
}

E. Vasya and a Tree

题意：给定一颗树，进行m个操作，每次将节点v子树中向下d+1层，的点全部加x，操作完成后询问每个点的值。

做法：dfs这棵树的同时，树状数组维护对应深度的影响，退出递归时，还原现场即可，类似于树上逆序对，因为操作的总和为m所以复杂度有保证。kd-tree和二维树状数组，都没卡过去。。。

#include <bits/stdc++.h>
#define pb push_back
#define fr first
#define sc second
#define P pair<int,ll>
typedef long long ll;
const int N = 300100;
using namespace std;
int n, m;
int dep[N],MX;
vector<int> G[N];
vector< P > A[N];
ll B[N], ans[N<<1];
void add(int x,ll v) {
    x += 10;
    for(int i=x;i;i-=(i&-i)) B[i] += v;
}
ll ask(int x) {
    ll ans = 0;
    x += 10;
    for(int i = x; i <= MX+20; i+=(i&(-i))) ans += B[i];
    return ans;
}
void dfs(int u,int fa) {
	dep[u] = dep[fa] + 1;
	MX = max(dep[u],MX);
	for(int i = 0; i < G[u].size(); ++i) {
		int v = G[u][i];
		if(v != fa) dfs(v,u);
	}
}
void dfs2(int u,int fa) {
    for(int i = 0; i < A[u].size(); ++i) add(A[u][i].fr,A[u][i].sc);
    ans[u] = ask(dep[u]);
	for(int i = 0; i < G[u].size(); ++i) {
		int v = G[u][i];
		if(v != fa) {
			dfs2(v,u);
		}
	}
	for(int i = 0; i < A[u].size(); ++i) add(A[u][i].fr,-A[u][i].sc);
}
int main() {
	scanf("%d",&n);
	for(int i = 1; i <= n-1; ++i) { int u,v;
		scanf("%d%d",&u,&v);
		G[u].pb(v); G[v].pb(u);
	}
	dep[0] = -1;
	dfs(1,0);
	scanf("%d",&m);
	for(int i = 1; i <= m; ++i) { int v,d; ll x;
		scanf("%d%d%lld",&v,&d,&x);
		A[v].pb(P(min(dep[v]+d,MX),x));
	}
	dfs2(1,0);
	for(int i = 1; i <= n; ++i)
		printf("%lld ",ans[i]);puts("");
	return 0;
}

F. Summer Practice Report

题意：有(n)页纸，第(i)页包含(a[i])个(T), (b[i])个(F)，要求将所有的(n)页纸并起来后，不能有连续的(k)个(T)或(F)，问是否有解。

做法：贪心构造dp。(dp[i][0/1]) 表示前(i)页纸放完，最后几个字符是(T)或(F)时，(T)或(F)最小的数目。如果(min(dp[n][0], dp[n][1]) <= k) 则满足条件。考虑如何(dp)，设上一张末尾的(T)有(pa)张或(F)有(pb)张，当前这一张有(a)个(T)，(b)个(F)。
先确定(dp[i][0])的转移，考虑放满(T)然后向其中插入(F)，用(pa)更新答案，那么如果(pa<=k)时，(num)即是需要插入的最少的(F)的个数，如果(b == num)， 那么用最后剩下的(T)更新答案，同时可以知道(b)的上界就是每个(T)之间都插入(k)个(F)，如果(b>num)且(b <= k*a)，就可以在最后一个(T)之前插入一个(F)，使得答案为(1)。用(pb)更新答案，思路类似，需要修改一下限制条件。(dp[i][1]) 也可以同样的转移。注意过程中会爆(int)

这道题在(dp)的同时贪心的转移，感觉思路十分清奇，看懂官方题解感觉自己dp烂的不要不要的。。。

#include <bits/stdc++.h>
typedef long long ll;
const int N = 300000+5;
const ll inf = 0x3f3f3f3f3f3f3f3f3f3f;
using namespace std;
int n, k, a[N], b[N];
int dp[N][2];
int cal(int pa, int pb, int a, int b) {
    ll ans = inf;
    if(pa <= k) {
        int num = (a + pa) / k + !!((a + pa) % k) - 1;
        if(b == num)
            ans = min(ans, pa + a - (ll)num*k);
        else if(b > num && (ll)b <= (ll)a*k)
            ans = min(ans, 1ll);
    }
    if(pb <= k) {
        int num = a / k + !!(a % k) - 1;
        if(b == num)
            ans = min(ans, a - (ll)num*k);
        else if(b > num && (ll)b <= (ll)(a-1)*k + (k-pb))
            ans = min(ans, 1ll);
    }
    return (int)ans;
}
int main() {
	scanf("%d %d", &n, &k);
	for(int i = 1; i <= n; ++i) scanf("%d", &a[i]);
	for(int i = 1; i <= n; ++i) scanf("%d", &b[i]);
	for(int i = 1; i <= n; ++i) dp[i][0] = dp[i][1] = inf;
	for(int i = 1; i <= n; ++i) {
        dp[i][0] = cal(dp[i-1][0], dp[i-1][1], a[i], b[i]);
        dp[i][1] = cal(dp[i-1][1], dp[i-1][0], b[i], a[i]);
	}
	if(dp[n][0] <= k || dp[n][1] <= k) puts("YES");
	else puts("NO");
	return 0;
}