C++ CSP-J 2022 星战 题解

#include <bits/stdc++.h>
using namespace std;
inline int read()
{
    long long x=0;
    bool flag=true;
    char ch=getchar();
    while(!isdigit(ch))
    {
        if(ch =='-') flag=false;
        ch=getchar();
    }
    while(isdigit(ch))
    {
        x=(x<<1)+(x<<3)+ch-'0';
        ch=getchar();
    }
    if(flag) return x;
    return ~(x - 1);
}
const int maxn =(int)5e5+5;
long long r[maxn],w[maxn],g[maxn];
int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);cout.tie(0);
    long long n=read(),m=read();
    std :: mt19937 rng(time(0));
    for (long long u=1; u<=n; u++) w[u]=rng();
    long long tar=std::accumulate(w+1, w+n+1,0LL);
    long long now=0;
    while (m--)
    {
        long long u= read(), v = read();
        r[v]+=w[u];
        g[v]=r[v];
        now+=w[u];
    }
    int q=read();
    while(q--)
    {
        long long t=read();
        if(t==1)
        {
            long long u=read(),v=read();
            r[v]-=w[u];
            now-=w[u];
        }
        else if(t==2)
        {
            long long v=read();
            now-=r[v];
            r[v]=0;
        }
        else if(t==3)
        {
            long long u=read(), v=read();
            r[v]+=w[u];
            now+=w[u];
        }
        else if(t==4)
        {
            long long v=read();
            now+=g[v]-r[v];
            r[v]=g[v];
        }
        if(now==tar) cout<<"YES"<<'\n';
        else cout<<"NO"<<'\n';
    }
    return 0;
}

#include <bits/stdc++.h>
using namespace std;
inline int read()
{
	long long x=0;
	bool flag=true;
	char ch=getchar();
	while(!isdigit(ch))
	{
		if(ch =='-') flag=false;
		ch=getchar();
	}
	while(isdigit(ch))
	{
		x=(x<<1)+(x<<3)+ch-'0';
		ch=getchar();
	}
	if(flag) return x;
	return ~(x - 1);
}
const int maxn =(int)5e5+5;
long long r[maxn],w[maxn],g[maxn];
int main()
{
	ios::sync_with_stdio(0);
	cin.tie(0);cout.tie(0);
	long long n=read(),m=read();
	std :: mt19937 rng(time(0));
	for (long long u=1; u<=n; u++) w[u]=rng();
	long long tar=std::accumulate(w+1, w+n+1,0LL);
	long long now=0;
	while (m--)
	{
		long long u= read(), v = read();
		r[v]+=w[u];
		g[v]=r[v];
		now+=w[u];
	}
	int q=read();
	while(q--)
	{
		long long t=read();
		if(t==1)
		{
			long long u=read(),v=read();
			r[v]-=w[u];
			now-=w[u];
		}
		else if(t==2)
		{
			long long v=read();
			now-=r[v];
			r[v]=0;
		}
		else if(t==3)
		{
			long long u=read(), v=read();
			r[v]+=w[u];
			now+=w[u];
		}
		else if(t==4)
		{
			long long v=read();
			now+=g[v]-r[v];
			r[v]=g[v];
		}
		if(now==tar) cout<<"YES"<<'\n';
		else cout<<"NO"<<'\n';
	}
	return 0;
}

请求三连

请求三连