Submission #586425

#	Time	Username	Problem	Language	Result	Execution time	Memory
586425		Tekor	Friend (IOI14_friend)	C++17	16 / 100	6 ms	7508 KiB

friend

#include "friend.h"
#include <bits/stdc++.h>
using namespace std;
#define pii pair <int,int>
#define f first
#define s second
#define mp make_pair
#define pb push_back
#define all(v) v.begin(),v.end()
#define ll long long
const int N = 3e5 + 100;
const int M = 1e3 + 100;
const ll INF = (ll)1e9 + 7ll;
vector <int> g[N];
bool con[M][M],u[N];
ll cost[N],dp[N][2][2],d[N];
void dfs(int v) {
	u[v] = 1;
	int kol = 0;
	for(auto to : g[v]) {
		if(u[to])continue;
		dfs(to);
		kol++;
	}
	if(kol == 0) {
		dp[v][1][1] = cost[v];
		dp[v][0][0] = 0;
		dp[v][1][0] = -INF;
		dp[v][0][1] = -INF;
		return;
	}
	for(int tt = 0;tt < 2;tt++) {
		dp[v][0][tt] = 0;
		dp[v][1][tt] = cost[v];
	}
	for(auto to : g[v]) {
		if(u[to])continue;
		for(int tt = 0;tt < 2;tt++) {
			dp[v][0][tt] += max(dp[to][0][tt],dp[to][1][tt]);
			dp[v][1][tt] += dp[to][0][tt];
		}
	}
}
int findSample(int n,int c[],int st[],int type[]){
	int mx = 0;
	bool ch1 = 0;
	for(int i = 0;i < n;i++) {
		mx = max(mx,c[i]);
		if(i > 0 && type[i] == 2)ch1 = 1;
	}
	if(mx == 1 && !ch1) {
		cost[0] = c[0];
		for(int i = 1;i < n;i++) {
			cost[i] = c[i];
			if(type[i] == 0) {
				g[st[i]].pb(i);
				g[i].pb(st[i]);	
			}else {
				for(auto to : g[st[i]]) {
					g[to].pb(i);
					g[i].pb(to);
				}
			}
		}
		int ans =0;
		for(int i = 0;i < n;i++) {
			if(!u[i]) {
				dfs(i);
				ans += max(dp[i][0][1],dp[i][1][0]);
			}
		}
		return ans;
	}
	if(n <= 10) {
		for(int i = 1;i < n;i++) {
			if(type[i] == 0) {
				g[st[i]].pb(i);
				g[i].pb(st[i]);
				con[i][st[i]] = 1;
				con[st[i]][i] = 1;
			}else if(type[i] == 1) {
				for(auto to : g[st[i]]) {
					g[to].pb(i);
					g[i].pb(to);
					con[i][to] = 1;
					con[to][i] = 1;
				}
			}else {
				g[st[i]].pb(i);
				g[i].pb(st[i]);
				con[i][st[i]] = 1;
				con[st[i]][i] = 1;
				for(auto to : g[st[i]]) {
					g[to].pb(i);
					g[i].pb(to);
					con[i][to] = 1;
					con[to][i] = 1;
				}
			}
		}
		int mx = 0;
		for(int i = 0;i < (1 << n);i++) {
			bool ch = 0;
			int sum = 0;
			for(int j = 0;j < n;j++) {
				if(i & (1 << j)) {
					for(int h = 0;h < j;h++) {
						if((i & (1 << h)) && con[j][h]) {
							ch = 1;
							break;
						}
					}
					if(ch)break;
					sum += c[j];
				}
			}
			if(!ch)mx = max(mx,sum);
		}
		return mx;
	}
	int ans = 0;
	if(type[1] == 1) {
		for(int i = 0;i < n;i++)ans += c[i];
	}else if(type[1] == 2){
		for(int i = 0;i < n;i++)ans = max(ans,c[i]);
	}
	return ans;
}

• Subtask 1: 0 / 11
• Subtask 2: 8 / 8
• Subtask 3: 8 / 8
• Subtask 4: 0 / 19
• Subtask 5: 0 / 23
• Subtask 6: 0 / 31