Codeforces Round 933 (Div. 3)

D. Rudolf and the Ball Game

set模拟 

#include<iostream>
#include<algorithm>
#include<cstring>
#include<vector>
#include<set>
using namespace std;
typedef long long ll;
const int N=2e5+10;
set<int>s1,s2;
int main()
{
	int t;cin>>t;
	while(t--)
	{
		int n,m,k;cin>>n>>m>>k;
		s1.clear();
		s2.clear();
		s1.insert(k);
		while(m--)
		{
			int num;cin>>num;
			char x;cin>>x;
			for(auto it:s1)
			{
				int num1=(num+it)%n;
				int num2=(it-num+n)%n;
				if(num1==0) num1=n;
				if(num2==0) num2=n;
				if(x=='0') s2.insert(num1);
				else if(x=='1') s2.insert(num2);
				else 
				{
					s2.insert(num1);
					s2.insert(num2);
				}
			}
			s1.clear();
			for(auto it:s2)
			{
				s1.insert(it);
			}
			s2.clear();
		}
		cout<<s1.size()<<endl;
		for(auto it:s1) cout<<it<<" ";
		cout<<endl;
		
	}
	return 0;
}

 dp直接遍历 看是否存在

#include<iostream>
#include<algorithm>
#include<cstring>
#include<vector>
#include<set>
using namespace std;
typedef long long ll;
const int N=2e5+10,M=1e3+10;
bool f[M][M];
int main()
{
	int t;cin>>t;
	while(t--)
	{
		int n,m,k;cin>>n>>m>>k;
		memset(f,0,sizeof f);
		f[0][k-1]=1;
	    for(int i=1;i<=m;i++)
	    {
	    	int num;cin>>num;
	    	char x;cin>>x;
	    	if(x=='0')//顺时针
	    	{
	    		for(int j=0;j<n;j++)
	    		{
	    			//顺时针到j则逆时针找回去与之对应的
	    			f[i][j]|=f[i-1][(j-num+n)%n];
				}
			}
			else if(x=='1')//逆时针
			{
				for(int j=0;j<n;j++)
				{
					//逆时针过来 则顺时针找回去
					f[i][j]|=f[i-1][(j+num)%n];
				}
			}
			else 
			{
				for(int j=0;j<n;j++)
				{
					f[i][j]|=f[i-1][(j+num)%n];
					f[i][j]|=f[i-1][(j-num+n)%n];
				}
			}
		}
		int sum=0;
		for(int j=0;j<n;j++)
		{
			if(f[m][j]) sum++;
		}
		cout<<sum<<endl;
		for(int j=0;j<n;j++)
		{
			if(f[m][j]) cout<<j+1<<" ";
		}
		cout<<endl;
	}
	return 0;
}