「CF471B」MUH and Important Things
It’s time polar bears Menshykov and Uslada from the zoo of St. Petersburg and elephant Horace from the zoo of Kiev got down to business. In total, there are n tasks for the day and each animal should do each of these tasks. For each task, they have evaluated its difficulty. Also animals decided to do the tasks in order of their difficulty. Unfortunately, some tasks can have the same difficulty, so the order in which one can perform the tasks may vary.
Menshykov, Uslada and Horace ask you to deal with this nuisance and come up with individual plans for each of them. The plan is a sequence describing the order in which an animal should do all the n tasks. Besides, each of them wants to have its own unique plan. Therefore three plans must form three different sequences. You are to find the required plans, or otherwise deliver the sad news to them by stating that it is impossible to come up with three distinct plans for the given tasks.
The first line contains integer n (1 ≤ n ≤ 2000) — the number of tasks. The second line contains n integers h1, h2, …, hn (1 ≤ hi ≤ 2000), where hi is the difficulty of the i-th task. The larger number hi is, the more difficult the i-th task is.
In the first line print “YES” (without the quotes), if it is possible to come up with three distinct plans of doing the tasks. Otherwise print in the first line “NO” (without the quotes). If three desired plans do exist, print in the second line n distinct integers that represent the numbers of the tasks in the order they are done according to the first plan. In the third and fourth line print two remaining plans in the same form.
If there are multiple possible answers, you can print any of them.
1 2 |
4 1 3 3 1 |
1 2 3 4 |
YES 1 4 2 3 4 1 2 3 4 1 3 2 |
1 2 |
5 2 4 1 4 8 |
1 |
NO |
In the first sample the difficulty of the tasks sets one limit: tasks 1 and 4 must be done before tasks 2 and 3. That gives the total of four possible sequences of doing tasks : [1, 4, 2, 3], [4, 1, 2, 3], [1, 4, 3, 2], [4, 1, 3, 2]. You can print any three of them in the answer.
In the second sample there are only two sequences of tasks that meet the conditions — [3, 1, 2, 4, 5] and [3, 1, 4, 2, 5]. Consequently, it is impossible to make three distinct sequences of tasks.
题解
问是否存在三种不同的不降排列
排序后贪心
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 |
#include<iostream> #include<cstdio> #include<cstring> #include<cstdlib> #include<algorithm> #include<cmath> #include<queue> #include<set> #include<map> #define pa pair<int,int> #define inf 1000000000 #define ll long long using namespace std; inline int read() { int x=0,f=1;char ch=getchar(); while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();} while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();} return x*f; } int n,tot; struct data{int v,pos;}a[2005]; bool operator<(data a,data b) { return a.v<b.v; } int main() { n=read(); for(int i=1;i<=n;i++)a[i].v=read(); for(int i=1;i<=n;i++)a[i].pos=i; sort(a+1,a+n+1); for(int i=1;i<=n;i++)if(a[i].v==a[i-1].v)tot++; if(tot<2)puts("NO"); else { puts("YES"); for(int i=1;i<=n;i++)printf("%d ",a[i].pos);puts(""); for(int i=1;i<=n;i++) if(a[i].v==a[i-1].v) { swap(a[i],a[i-1]);break; } for(int i=1;i<=n;i++)printf("%d ",a[i].pos);puts(""); for(int i=n;i;i--) if(a[i].v==a[i-1].v) { swap(a[i],a[i-1]);break; } for(int i=1;i<=n;i++)printf("%d ",a[i].pos);puts(""); } return 0; } |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 |
#include<iostream> #include<cstdio> #include<cstring> #include<cstdlib> #include<algorithm> #include<cmath> #include<queue> #include<set> #include<map> #define pa pair<int,int> #define inf 1000000000 #define ll long long using namespace std; inline ll read() { ll x=0,f=1;char ch=getchar(); while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();} while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();} return x*f; } int n,m,ans; int next[200005]; int a[200005],b[200005]; int main() { n=read();m=read(); for(int i=1;i<=n;i++) a[i]=read(); for(int i=1;i<=m;i++) b[i]=read(); for(int i=1;i<n;i++)a[i]=a[i+1]-a[i]; for(int i=1;i<m;i++)b[i]=b[i+1]-b[i]; int j=0; for(int i=2;i<n;i++) { while(j>0&&a[j+1]!=a[i])j=next[j]; if(a[j+1]==a[i])j++; next[i]=j; } j=0; for(int i=2;i<n;i++) { while(j>0&&a[j+1]!=b[i])j=next[j]; if(a[j+1]==b[i])j++; if(j==m-1)ans++; } printf("%d",ans); return 0; } |