传送门
Examples
input
3
3 1 2 3
2 1 5
3 2 4 6
2
3 2 3
3 2 2
output
2 2 3
input
3
3 1 2 3
2 1 5
3 2 4 6
2
3 2 3
2 2 3
output
1 2 3
input
3
3 1 2 3
2 1 5
3 2 4 6
2
3 2 3
2 2 3
output
3 2 2
input
4
1 10
1 4
1 7
1 3
0
output
1 1 1 1
大致翻译:
大致题意:给定数组个数n,输出每组数据的个数和具体数据(从小到大排列)给m组不同的组合情况,使得在之后求解中这m个组合被限制,求未被限制的最大的组合情况,并输出此组合;
思路:(我不会……问学长的……)先看最大的情况有没有被限制若无则直接输出,若被限制则在限制的几种组合情况中退一个位置寻找最大的组合情况并输出;
代码:
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2e5+10;
int n,m,maxx,num[N]/*每组数据个数*/,sum[N];/*数据的总和*/
vector<int>a[N]/*具体数据*/,lim[N];/*限制数据*/
map<vector<int>,int>mp;
void MAX(){
vector<int>b;
for(int i=0;i<n;i++)b.push_back(num[i]-1);
if(mp.find(b)!=mp.end())return ;
for(auto it:b)cout<<it+1<<' ';
exit(0);
}
void find(vector<int>b,int sum){//除了限制之外的找最大
for(int i=0;i<n;i++){
b[i]--;//退一个位置
if(mp.find(b)==mp.end()&&b[i]>=0)
maxx=max(maxx,sum-a[i][b[i]+1]+a[i][b[i]]);
b[i]++;
}
}
void scan(vector<int>b,int sum){
for(int i=0;i<n;i++){
b[i]--;
if(mp.find(b)==mp.end()&&b[i]>=0&&maxx==sum-a[i][b[i]+1]+a[i][b[i]]){
for(auto it:b)cout<<it+1<<' ';
exit(0);
}
b[i]++;
}
}
int main(){
ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
cin>>n;
for(int i=0;i<n;i++){
cin>>num[i];
for(int j=0;j<num[i];j++){
int x;cin>>x;a[i].push_back(x);
}
}
cin>>m;
for(int i=0;i<m;i++){
for(int j=0;j<n;j++){
int x;cin>>x;x--;
lim[i].push_back(x);
sum[i]+=a[j][x];
}
mp[lim[i]]++;
}
MAX();//找最大,若最大存在则说明在最大中某一位退一个位置为最大
for(int i=0;i<m;i++)find(lim[i],sum[i]);
for(int i=0;i<m;i++)scan(lim[i],sum[i]);
return 0;
}