bzoj4419 [Shoi2013]发微博

题目

一道stl应用题目，只要用set模拟一下就可以了。

#include<bits/stdc++.h>
#define N 200000
using namespace std;
set <int> E[N+1];
set <int> ::iterator it;
int Ans[N+1],n,m,x,y;
int give[N+1];
char opt;
int read()
{
    int x=0;
    char c=getchar();
    while(c<'0'||c>'9')c=getchar();
    while(c<='9'&&c>='0')x=x*10+c-'0',c=getchar();
    return x;
}
int main()
{
   // freopen("in.txt","r",stdin);
    n=read(),m=read();
    for(int i=1;i<=m;i++)
    {
        opt=getchar();
        if(opt=='!')
        {
            x=read();
            give[x]++;
        }
        if(opt=='+')
        {
            x=read(),y=read();
            E[x].insert(y);
            E[y].insert(x);
            Ans[x]-=give[y],Ans[y]-=give[x];
        }
        if(opt=='-')
        {
            x=read(),y=read();
            E[x].erase(y);
            E[y].erase(x);
            Ans[x]+=give[y],Ans[y]+=give[x];
        }       
    }
    for(int i=1;i<=n;i++)
        for(it=E[i].begin();it!=E[i].end();it++)
            Ans[*it]+=give[i];
    printf("%d",Ans[1]);
    for(int i=2;i<=n;i++)printf(" %d",Ans[i]);
    return 0;
}