Konrad and Company Evaluation

Posted 2022-12-10 cadcadcad

F. Konrad and Company Evaluation

参考：[codeforces 1230F]Konrad and Company Evaluation-暴力

思路：题意分析见参考博客。因为求的是三元组的个数，所以在保存的时候的时候就保存为有向图，让工资少的员工指向工资多的员工，那么求三元组的时候，只需要以三元组中间的那个员工为参考点来求解即可。

ll cal(int i)

    return (cnt[i]-g[i].size())*g[i].size();

那么当进行操作的时候，因为每一次会加n所以肯定会比之间它所连接的两个点要大，那么对于该点指向的那个点来说，组成的三元组就要重新进行计算了，所以先把之前算过的部分减去，再把新的部分加上。

代码：

// Created by CAD on 2019/10/1.
#include <bits/stdc++.h>
#define ll long long
using namespace std;

const int maxn=1e6+5;
int cnt[maxn];
vector<int> g[maxn];
ll cal(int i)

    return (cnt[i]-g[i].size())*g[i].size();

int main()

    ios::sync_with_stdio(false);
    cin.tie(0);
    int n,m;    cin>>n>>m;
    for(int i=1,u,v;i<=m;++i)
    
        cin>>u>>v;
        if(u>v) swap(u,v);
        g[u].push_back(v);
        cnt[u]++,cnt[v]++;
    
    ll ans=0;
    for(int i=1;i<=n;++i)   ans+=cal(i);
    int q;  cin>>q;
    cout<<ans<<endl;
    while(q--)
    
        int x;  cin>>x;
        ans-=cal(x);
        for(auto i:g[x])
        
            ans-=cal(i);
            g[i].push_back(x);
            ans+=cal(i);
            g[x].clear();
        
        cout<<ans<<endl;
    
    return 0;