D – Happy Birthday!
题目描述:
n个数字,问能不能选出两个不同的序列使得序列和模200后相同,从小到大输出两个序列每个数在原数组的下标
思路:
考虑鸽巢原理,只要构造出大于200个的不同的序列,就可以通过鸽巢原理得到两个序列
所以考虑二进制枚举
8位的二进制就已经有255了,大于200,所以只要数量大于8就绝对可以构造出两个不同的序列
小于8的情况就求出所有的判断就行
/*
Work by: Chelsea
Problem: 不知名屑题
Knowledge: 垃圾算法
Time: O(能过)
*/
#include<map>
#include<set>
#include<stack>
#include<queue>
#include<cmath>
#include<bitset>
#include<cstdio>
#include<string>
#include<vector>
#include<sstream>
#include<cstring>
#include<stdlib.h>
#include<iostream>
#include<algorithm>
//#include<unordered_map>
using namespace std;
#define endl '\n'
#define inf 0x3f3f3f3f
#define INF 0x3f3f3f3f3f3f3f3f
#define mod 1000000007
#define y1 y114514
#define sz(s) (int)(s).size()
#define sd(n) scanf("%d",&n)
#define sl(n) scanf("%lld",&n)
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define sdd(n,m) scanf("%d %d",&n,&m)
#define io ios::sync_with_stdio(false);
#define mem(a,b) memset((a),(b),sizeof(a))
#define m_p(a,b) make_pair(a, b)
//#define int long long
typedef long long ll;
typedef pair <int,int> pii;
typedef unsigned long long ull;
//不开longlong见祖宗!
//不改范围见祖宗!!!
#define MAX 300000 + 50
int n, m, k;
int x, y, z;
ll a, b, c;
string s, t;
char op;
int tr[MAX];
vector<int>ans[300];
void work(){
cin>>n;
rep(i, 1, n)cin>>tr[i];
int num = min(8, n);
vector<int>v;v.push_back(0);
for(int i = 1; i <= (1 << num) - 1; ++i){
bitset<10>b(i);
int p = 0;
for(int j = 0; j < num; ++j){
if(b[j]){
ans[i].push_back(j + 1);
p += tr[j + 1];
}
p %= 200;
}
v.push_back(p);
}
rep(i, 1, sz(v) - 1){
rep(j, i + 1, sz(v) - 1){
if(v[i] == v[j]){
// cout<<i<<' '<<j<<endl;
cout<<"Yes\n";
cout<<ans[i].size();
rep(k, 0, sz(ans[i]) - 1)cout<<' '<<ans[i][k];
cout<<endl;
cout<<ans[j].size();
rep(k, 0, sz(ans[j]) - 1)cout<<' '<<ans[j][k];
cout<<endl;
return;
}
}
}
cout<<"No\n";
}
int main(){
io;
// int tt;cin>>tt;
// for(int _t = 1; _t <= tt; ++_t){
// printf("Case #%d: ", _t);
work();
// }
return 0;
}