codeforces 191 C. Fools and Roads(樹上差分)

題目鏈接:https://codeforc.es/contest/191/problem/C

思路:樹上差分模板

對於每一次修改s,t,將s,t的權+1;將LCA(s,t)和father[LCA(s,t)]的權-1;一個點最終被覆蓋的次數就是這個點所在子樹的權和。(這是對於點的差分),本題是關於邊的,與點的差別就是將LCA(s,t)的權-2,且不減father[LCA(s,t)]的權

#include <cstdio>
#include <cstdlib>
#include <cassert>
#include <cstring>
#include <bitset>
#include <cmath>
#include <cctype>
#include <iostream>
#include <algorithm>
#include <string>
#include <vector>
#include <queue>
#include <map>
#include <set>
#include <sstream>
#include <iomanip>
using namespace std;
typedef long long ll;
typedef vector<long long> VI;
typedef unsigned long long ull;
const ll inff = 0x3f3f3f3f3f3f3f3f;
#define FOR(i,a,b) for(int i(a);i<=(b);++i)
#define FOL(i,a,b) for(int i(a);i>=(b);--i)
#define SZ(x) ((long long)(x).size())
#define REW(a,b) memset(a,b,sizeof(a))
#define inf int(0x3f3f3f3f)
#define si(a) scanf("%d",&a)
#define sl(a) scanf("%lld",&a)
#define sd(a) scanf("%lf",&a)
#define ss(a) scanf("%s",a)
#define mod ll(1e9+7)
#define pb push_back
#define eps 1e-6
#define lc d<<1
#define rc d<<1|1
#define Pll pair<ll,ll>
#define P pair<int,int>
#define pi acos(-1)
int n,k,x,y,fa[100008][20],deep[100008],d[100008],ans[100008],zz;
vector<P>g[100008];
void dfs(int u,int faa)
{
    for(int i=0;i<g[u].size();i++)
    {
        int v=g[u][i].first;
        if(v!=faa)  deep[v]=deep[u]+1,dfs(v,u),fa[v][0]=u;
    }
}
void bz()
{
    FOR(i,1,18)
     FOR(j,1,n)
      fa[j][i]=fa[fa[j][i-1]][i-1];
}
int lca(int x,int y)
{
    if(deep[x]<deep[y])  swap(x,y);
    int d=deep[x]-deep[y];
    FOR(i,0,18)
     if(d&(1<<i))  x=fa[x][i];
    if(x==y)  return x;
    FOL(i,18,0)
     if(fa[x][i]!=fa[y][i])  x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}
int dfs1(int u,int fa)
{
    int qw=d[u];
    for(int i=0;i<g[u].size();i++)
    {
        int v=g[u][i].first;
        if(v==fa)  continue;
        ans[g[u][i].second]=dfs1(v,u);
        qw+=ans[g[u][i].second];
    }
    return qw;
}
int main()
{
    cin.tie(0);
    cout.tie(0);
    cin>>n;
    FOR(i,1,n-1)
    {
        si(x),si(y);
        g[x].pb(P(y,i));
        g[y].pb(P(x,i));
    }
    dfs(1,0),bz();
    cin>>k;
    while(k--)
    {
        si(x),si(y);
        zz=lca(x,y);
        d[x]++,d[y]++,d[zz]-=2;
    }
    dfs1(1,0);
    FOR(i,1,n-1) printf("%d ",ans[i]);
    puts("");
    return 0;
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章