Codeforces 5A Chat Servers Outgoing Traffic set的使用

水过~


#include 
#include 
#include 
using namespace std;

int nc=0;
set se;

int main()
{
	string s;
	int ans=0;
	while( getline(cin,s) )
	{
		if(s[0]=='+')
		{
			if(!se.count(s.substr(1)))
			{
				se.insert(s.substr(1));
				nc++;
			}
		}
		else if(s[0]=='-')
		{
			if(se.count(s.substr(1)))
			{
				se.erase(s.substr(1));
				nc--;
			}
		}
		else
		{
			int pos=s.find_first_of(':');
			string name=s.substr(0,pos);
			if(se.count(name))
			{
				string mes=s.substr(pos+1);
				ans+=mes.size()*nc;
			}
		}
	}
	cout << ans << endl;
	return 0;
}


你可能感兴趣的:(Codeforces)