Submission #39055

#	Time	Username	Problem	Language	Result	Execution time	Memory
39055		Scayre	Marriage questions (IZhO14_marriage)	C++14	14 / 100	1500 ms	3944 KiB

marriage

//
// omae wa mou shindeiru.
//

#pragma GCC optimize("Ofast")
#pragma GCC target("sse,sse2,sse3,sse3,sse4,popcnt,abm,mmx")

#include <map>
#include <set>
#include <list>
#include <cmath>
#include <ctime>
#include <deque>
#include <queue>
#include <stack>
#include <string>
#include <bitset>
#include <cstdio>
#include <limits>
#include <vector>
#include <climits>
#include <cstring>
#include <cstdlib>
#include <fstream>
#include <numeric>
#include <sstream>
#include <cassert>
#include <iomanip>
#include <iostream>
#include <algorithm>

#define ll long long
#define ull unsigned ll
#define ioi exit(0);

#define f first
#define s second

#define inf (int)1e9 + 7

#define NFS ios_base :: sync_with_stdio(0), cin.tie(0), cout.tie(0);

#define mp make_pair

#define lb(x) lower_bound(x)
#define ub(x) upper_bound(x)

#define pb push_back
#define ppb pop_back

#define endl "\n"

#define in(x) insert(x)

#define sz(x) (int)x.size()

#define all(x) x.begin(),x.end()

#define pw2(x) (1<<x) //2^x

#define sqr(x) ((x) * 1ll * (x))

using namespace std;

const int N = (int)3e4 + 7, MOD = (int)1e9 + 7;

int n,k,m;

vector <int> a[N];

bool ok=0;

map <int,vector<int> > mpp;

ll ans,res;

set <int> s;

void calc(int pos,int cnt){
	if(pos>n){
		return;
	}
	if(s.size()==m){
		ans=max(ans,(ll)n-pos+1);
		return;
	}
	for(int i=0;i<a[pos].size();i++){
		s.insert(a[pos][i]);
		mpp[a[pos][i]].pb(cnt);
		calc(pos+1,cnt+1);
		for(auto it=s.begin();it!=s.end();it++){
			int z=*it;
			if(mpp[z][mpp[z].size()-1]>=cnt){
				mpp[z].ppb();
				if(mpp[z].size()==0)s.erase(z);
			}
		}
	}
}

int main(){

	#ifdef IOI2019
		freopen ("in.txt", "r", stdin);
	#endif

	NFS

	cin >> n >> m >> k;

	for(int i=1;i<=k;i++){
		int x,y;
		cin >> x >> y;
		a[x].pb(y);
	}

	for(int i=1;i<=n;i++){
		ans=0;
		calc(i,1);
		res+=ans;
		if(ans==0){
			break;
		}
		s.clear();
		mpp.clear();
	}

	cout << res << endl;


	#ifdef IOI2019
		cout << "\nTime Elapsed : " << clock () * 1.0 / CLOCKS_PER_SEC << endl;
	#endif

	ioi
}

Compilation message (stderr)

marriage.cpp: In function 'void calc(int, int)':
marriage.cpp:83:13: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  if(s.size()==m){
             ^
marriage.cpp:87:15: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for(int i=0;i<a[pos].size();i++){
               ^

• Subtask 1: 14 / 100