CF1574D.The Strongest Build (bfs)

linkkkk
题意：
给出有\(n\)行的矩阵，每行有\(a_i\)个元素。保证按照升序排列。
每行都选择一个元素使得获得的价值最大，其中有\(m\)种不可以选的方案。
思路：
\(n<=10\)是个很重要的条件。
由于每行都选最后的元素是最优的，考虑怎么躲避开不可选的方案。贪心的考虑，如果一个方案不可选，那么将其中的一个元素变成选前面的一位元素是比较优的，直接\(O(nm)\)枚举就好了
代码：

// Problem: D. The Strongest Build
// Contest: Codeforces - Educational Codeforces Round 114 (Rated for Div. 2)
// URL: https://codeforces.com/contest/1574/problem/D
// Memory Limit: 256 MB
// Time Limit: 3000 ms
// 
// Powered by CP Editor (https://cpeditor.org)

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

const int maxn=2e5+10;

vector<int>ans,a[15],b[maxn];
map<vector<int>,int>mp;

int main(){
	int n;cin>>n;
	for(int i=0;i<n;i++){
		int m;cin>>m;
		ans.push_back(m-1);
		for(int j=0;j<m;j++){
			int x;cin>>x;
			a[i].push_back(x);
		}
	}
	int m;cin>>m;
	for(int i=0;i<m;i++){
		for(int j=0;j<n;j++){
			int x;cin>>x;
			x--;
			b[i].push_back(x);
		}
		mp[b[i]]=1;
	}
	if(mp.find(ans)==mp.end()){
		for(auto it:ans) cout<<it+1<<" ";
		puts("");
		return 0;
	}
	int res=0;
	for(int i=0;i<m;i++){
		int sum=0;
		for(int j=0;j<n;j++) sum+=a[j][b[i][j]];
		for(int j=0;j<n;j++){
			sum-=a[j][b[i][j]];
			if(b[i][j]==0) continue;
			b[i][j]--;
			sum+=a[j][b[i][j]];
			if(mp.find(b[i])==mp.end()){
				if(sum>res) ans=b[i],res=sum;
			}
			sum-=a[j][b[i][j]];
			b[i][j]++;
			sum+=a[j][b[i][j]];
		}
	}
	for(auto it:ans){
		cout<<it+1<<" ";
	}
	puts("");
	
	
	
	
	
	
	
	
	
	
	
	
	
	
	return 0;
}