Submission #579286

#	Time	Username	Problem	Language	Result	Execution time	Memory
579286		1ne	Paint By Numbers (IOI16_paint)	C++14	32 / 100	1 ms	596 KiB

paint

#include "paint.h"
#include<bits/stdc++.h>
using namespace std;

std::string solve_puzzle(std::string s, std::vector<int> c) {
    int n = (int)s.length();
    int k = (int)c.size();
    vector<vector<vector<int>>>dp(n + 1,vector<vector<int>>(k + 1,vector<int>(2,-1)));
    function<int(int,int,int)>solve = [&](int u,int v,bool ok){
		if (u == n){
			if (v != k)return dp[u][v][ok] = 0;
			return dp[u][v][ok] = 1;
		}
		if (dp[u][v][ok]!=-1)return dp[u][v][ok];
		if (v == k){
			return dp[u][v][ok] = solve(u + 1,v,0);
		}
		int ans = 0 ;
		if (u + c[v]<= n && !ok){
			ans += solve(u + c[v],v + 1,1);
		}
		ans +=solve(u + 1,v,0);
		return dp[u][v][ok] = ans;
	 };
	/* dp[0][0][0] = 1;
	 for (int i = 0;i<n;++i){
		for (int j = 0;j<=k;++j){
			dp[i + 1][j][0] +=dp[i][j][1] + dp[i][j][0];
			if (j < k && i + c[j] <=n){
				dp[i + c[j]][j + 1][1]+=dp[i][j][0];
			}
		}
	 }
	 for (int i = 0;i<=n;++i){
		for (int j = 0;j<=k;++j){
			for (int p = 0;p<2;++p){
				cout<<dp[i][j][p]<<" ";
			}
			cout<<'\n';
		}
		cout<<'\n';
	 }
	 */
    solve(0,0,0);
    string answer(n,'?');
    vector<int>all(n,0);
    for (int i = 1;i<=k;++i){
		 vector<int>ans(n,0);
		 int total = 0;
		 for (int j = 0;j<=n;++j){
			if (j - c[i - 1] >= 0){
				for (int p = 1;p<=c[i - 1];++p){
					if (dp[j][i][1]!= -1){
						ans[j - p]+=dp[j][i][1];
						all[j - p]+=dp[j][i][1];
					}
				}
			}
			if (dp[j][i][1]!=-1)
				total+=dp[j][i][1];
		 }
		 for (int j = 0;j<n;++j){
			 if (ans[j] == total)answer[j] = 'X';
		 }
	 }
	 for (int i = 0;i<n;++i){
		if (all[i] == 0){
			answer[i] = '_';
		} 
	 }
	 return answer;
}

• Subtask 1: 7 / 7
• Subtask 2: 3 / 3
• Subtask 3: 22 / 22
• Subtask 4: 0 / 27
• Subtask 5: 0 / 21
• Subtask 6: 0 / 10
• Subtask 7: 0 / 10